diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-07 02:07:23 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-07 02:07:23 +0200 |
| commit | b7b305a6dda37e0ac131d42007c4014e82decd2c (patch) | |
| tree | 52b20357630152cea4cedfd9a0d356112fd59196 /models | |
| parent | a438c6e88484a68abdd17384e720452ffa2b96bb (diff) | |
augment paths
Diffstat (limited to 'models')
| -rw-r--r-- | models/base_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
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 |