fix Deepphy's motion model issue

chenyouxin113 · Jul 30, 2021 · f137edb · f137edb
1 parent 3481383
commit f137edb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/nets/models/sub_models/MotionModel.py b/nets/models/sub_models/MotionModel.py
@@ -28,7 +28,8 @@ def forward(self, inputs, mask1, mask2):
         M1 = torch.tanh(self.m_batch_Normalization1(self.m_conv1(inputs)))
         M2 = self.m_batch_Normalization2(self.m_conv2(M1))
         # element wise multiplication Mask1
-        g1 = torch.tanh(torch.mul(torch.ones(size= mask1.shape).to('cuda')@mask1, M2))
+        ones = torch.ones(size=M2.shape).to('cuda')
+        g1 = torch.tanh(torch.mul(ones@mask1, M2))
         M3 = self.m_dropout1(g1)
         # pooling
         M4 = self.m_avg1(M3)

diff --git a/params.json b/params.json
@@ -1,7 +1,7 @@
 {
     "__TIME__" : 1,
     "__PREPROCESSING__" : 0,
-    "__MODEL_SUMMARY__" : 1,
+    "__MODEL_SUMMARY__" : 0,
     "options":{
         "parallel_criterion" : 1,
         "parallel_criterion_comment" : "TODO need to verification"
@@ -38,7 +38,7 @@
                     "adam","sgd","rms_prop","ada_delta","ada_grad","ada_max",
                     "ada_mw","a_sgd","lbfgs","n_adam","r_adam","rprop","sparse_adam"
                 ],
-        "learning_rate": 0.001,
+        "learning_rate": 1,
         "learning_rate_comment": [
             "DeepPhys : lr = 1",
             "PhysNet : lr = 0.001"