Fix
This commit is contained in:
parent
99395ca3ec
commit
aa2a248503
@ -69,8 +69,6 @@ class Loco:
|
|||||||
else:
|
else:
|
||||||
turning_model_path = "/home/beauvill/Repos/monoloco/data/outputs/casr-210615-1128.pkl"
|
turning_model_path = "/home/beauvill/Repos/monoloco/data/outputs/casr-210615-1128.pkl"
|
||||||
|
|
||||||
print('-'*10 + 'Output size :' + str(turning_output_size) + '-'*10)
|
|
||||||
|
|
||||||
if not device:
|
if not device:
|
||||||
self.device = torch.device('cpu')
|
self.device = torch.device('cpu')
|
||||||
else:
|
else:
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user