t

HuangYiran · Feb 1, 2022 · 0463d61 · 0463d61
1 parent 50b2c16
commit 0463d61
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 1 deletion.
diff --git a/__pycache__/experiment.cpython-38.pyc b/__pycache__/experiment.cpython-38.pyc
diff --git a/__pycache__/utils.cpython-38.pyc b/__pycache__/utils.cpython-38.pyc
diff --git a/models/crossatten/__pycache__/embedding.cpython-38.pyc b/models/crossatten/__pycache__/embedding.cpython-38.pyc
diff --git a/models/crossatten/embedding.py b/models/crossatten/embedding.py
@@ -474,7 +474,7 @@ def __init__(self, input_shape, token_d_model, dw_layers = 3 ,se_layers = 3):
         shape = self.get_the_shape(input_shape) # B C F L
 
         dim = shape[1]*shape[2]
-        print("------------dim-------------", shape[1], "    ", shape[1])
+        print("------------dim-------------", shape[1], "    ", shape[2])
         self.fc = nn.Linear(dim,int(dim/2))
         self.activation = nn.ReLU()