From b7b305a6dda37e0ac131d42007c4014e82decd2c Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 7 Jun 2018 02:07:23 +0200 Subject: augment paths --- models/base_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'models/base_model.py') diff --git a/models/base_model.py b/models/base_model.py index 9b55afe..d3d07d4 100644 --- a/models/base_model.py +++ b/models/base_model.py @@ -11,7 +11,7 @@ class BaseModel(): self.gpu_ids = opt.gpu_ids self.isTrain = opt.isTrain self.Tensor = torch.cuda.FloatTensor if self.gpu_ids else torch.Tensor - self.save_dir = os.path.join(opt.checkpoints_dir, opt.name) + self.save_dir = os.path.join(opt.checkpoints_dir, opt.module_name, opt.name) def set_input(self, input): self.input = input -- cgit v1.2.3-70-g09d2