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

Test

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

1
monoloco/network/net.py
View File

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

1
monoloco/predict.py
View File

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