diff --git a/train.py b/train.py
index e68f80c..de0a81f 100644
--- a/train.py
+++ b/train.py
@@ -42,6 +42,10 @@ def train(opt):
         torch.cuda.manual_seed(123)
     else:
         torch.manual_seed(123)
+
+    if not os.path.exists(opt.saved_path):
+        os.mkdir(opt.saved_path)
+
     output_file = open(opt.saved_path + os.sep + "logs.txt", "w")
     output_file.write("Model's parameters: {}".format(vars(opt)))
     training_params = {"batch_size": opt.batch_size,