add some update

wyddmw · May 5, 2021 · c27f6ba · c27f6ba
1 parent e450831
commit c27f6ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model/model.py b/model/model.py
@@ -38,7 +38,7 @@ def forward(self, x):
         x = F.relu(self.bn5(self.fc2(x)))
         x = self.fc3(x)
 
-        iden = Variable(torch.from_numpy(np.array([1,0,0,0,1,0,0,0,1]).astype(np.float32))).view(1,9).repeat(batchsize,1)
+        iden = Variable(torch.from_numpy(np.array([1,0,0,0,0,1,0,0,0,0,1,0,0,0,0,1]).astype(np.float32))).view(1,16).repeat(batchsize,1)
         if x.is_cuda:
             iden = iden.cuda()
         x = x + iden