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

fix assertion

Browse Source
This commit is contained in:
Lorenzo 2021-04-22 16:00:18 +02:00
parent e155100434
commit 7ac6855af4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
monoloco/train/trainer.py
View File

@ -71,7 +71,7 @@ class Trainer:
now_time = now.strftime("%Y%m%d-%H%M")[2:] now_time = now.strftime("%Y%m%d-%H%M")[2:]
name_out = name + '-' + now_time + '.pkl' name_out = name + '-' + now_time + '.pkl'
self.path_out = os.path.join(dir_out, name_out) self.path_out = os.path.join(dir_out, name_out)
assert dir_out, "Directory to save the model not found" assert os.path.exists(dir_out), "Directory to save the model not found"
print(self.path_out) print(self.path_out)
# Select the device # Select the device
use_cuda = torch.cuda.is_available() use_cuda = torch.cuda.is_available()