diff options
Diffstat (limited to 'models/base_model.py')
| -rw-r--r-- | models/base_model.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/models/base_model.py b/models/base_model.py index d3d07d4..4e7122a 100644 --- a/models/base_model.py +++ b/models/base_model.py @@ -12,6 +12,7 @@ class BaseModel(): 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.module_name, opt.name) + print(self.save_dir) def set_input(self, input): self.input = input |