Test

monoloco/network/net.py

@@ -96,6 +96,7 @@ class Loco:
 
             self.model.load_state_dict(torch.load(model_path, map_location=lambda storage, loc: storage))
             if casr:
+                print("WTF")
                 self.turning_model.load_state_dict(torch.load(turning_model_path,
                                                               map_location=lambda storage, loc: storage))
         else:

monoloco/predict.py

@@ -161,6 +161,7 @@ def predict(args):
 
     # Load Models
     if args.mode in ('mono', 'stereo'):
+        print(args.casr)
         net = Loco(
             model=dic_models[args.mode],
             mode=args.mode,