From 7c9a1a06ec90abb4bd29fe3d6a559440700a6cda Mon Sep 17 00:00:00 2001 From: icmpnorequest Date: Tue, 16 Jun 2020 14:28:02 +0800 Subject: [PATCH] Add saved_path dir checker --- train.py | 4 ++++ 1 file changed, 4 insertions(+) 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,