关闭多头输出的打印信息

Lix1993 · Jul 2, 2022 · 2f87151 · 2f87151
1 parent 007d33f
commit 2f87151
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/MyTransformer.py b/MyTransformer.py
@@ -5,7 +5,7 @@
 import copy
 import torch
 
-is_print_shape = True
+is_print_shape = False
 
 
 class MyTransformer(nn.Module):