From 36bdc6335f9363d9b2ad2bb35801218de9c0f2d9 Mon Sep 17 00:00:00 2001 From: Charles Joseph Pierre Beauville Date: Mon, 28 Jun 2021 01:08:34 +0200 Subject: [PATCH] Fixed for tests --- monoloco/network/net.py | 1 - monoloco/predict.py | 3 +-- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/monoloco/network/net.py b/monoloco/network/net.py index 2299b0a..d06c32b 100644 --- a/monoloco/network/net.py +++ b/monoloco/network/net.py @@ -96,7 +96,6 @@ 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 660ade7..0ea1008 100644 --- a/monoloco/predict.py +++ b/monoloco/predict.py @@ -132,7 +132,7 @@ def factory_from_args(args): if args.casr_std: args.casr = 'std' - else: + elif args.casr: args.casr = 'nonstd' # Patch for stereo images with batch_size = 2 @@ -161,7 +161,6 @@ def predict(args): # Load Models if args.mode in ('mono', 'stereo'): - print(args.casr) net = Loco( model=dic_models[args.mode], mode=args.mode,