from local

dohe0342 · Jan 16, 2024 · 741bb00 · 741bb00
1 parent e39cb69
commit 741bb00
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/scripts/whale/.test.py.swp b/scripts/whale/.test.py.swp
diff --git a/scripts/whale/test.py b/scripts/whale/test.py
@@ -5,7 +5,7 @@
 a = torch.randn(642, 768)
 b = torch.randn(56, 768)
 lm_am_sim_cp = torch.matmul(a, b)
-print(lm_am_sim_cp)
+print(lm_am_sim_cp.size())
 
 plt.matshow(lm_am_sim_cp.numpy())
 plt.colorbar()