From 7ac6855af412d54125488ac6affa740c85d39d7b Mon Sep 17 00:00:00 2001
From: Lorenzo <lorenzobertoni9@gmail.com>
Date: Thu, 22 Apr 2021 16:00:18 +0200
Subject: [PATCH] fix assertion

---
 monoloco/train/trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/monoloco/train/trainer.py b/monoloco/train/trainer.py
index 7051a9a..079d90c 100644
--- a/monoloco/train/trainer.py
+++ b/monoloco/train/trainer.py
@@ -71,7 +71,7 @@ class Trainer:
             now_time = now.strftime("%Y%m%d-%H%M")[2:]
             name_out = name + '-' + now_time + '.pkl'
             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)
         # Select the device
         use_cuda = torch.cuda.is_available()