charles/monoloco
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed net.py

Browse Source
This commit is contained in:
Charles Joseph Pierre Beauville 2021-06-28 01:21:30 +02:00
parent aa2a248503
commit 5870848da9
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
monoloco/network/net.py
View File

@ -100,8 +100,9 @@ class Loco:
self.model = model self.model = model
self.model.eval() # Default is train self.model.eval() # Default is train
self.model.to(self.device) self.model.to(self.device)
self.turning_model.eval() # Default is train if casr:
self.turning_model.to(self.device) self.turning_model.eval() # Default is train
self.turning_model.to(self.device)
def forward(self, keypoints, kk, keypoints_r=None): def forward(self, keypoints, kk, keypoints_r=None):
""" """