From f20e17709cd740ec30e5ed7c3e1bf851faaf049f Mon Sep 17 00:00:00 2001 From: Charles Joseph Pierre Beauville Date: Mon, 28 Jun 2021 01:04:35 +0200 Subject: [PATCH] Test --- monoloco/network/net.py | 3 ++- monoloco/predict.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/monoloco/network/net.py b/monoloco/network/net.py index 338f531..2299b0a 100644 --- a/monoloco/network/net.py +++ b/monoloco/network/net.py @@ -62,7 +62,7 @@ class Loco: print("CASR with standard gestures") turning_output_size = 3 turning_model_path = "/home/beauvill/Repos/monoloco/data/outputs/casr_standard-210613-0005.pkl" - elif casr== 'nonstd': + elif casr == 'nonstd': turning_output_size = 4 if casr_model: turning_model_path = casr_model @@ -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: diff --git a/monoloco/predict.py b/monoloco/predict.py index 57126f4..660ade7 100644 --- a/monoloco/predict.py +++ b/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,