From d95e6935985fda5e598601504a00e23a1b15d068 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 9 Oct 2019 16:25:04 -0400 Subject: [PATCH] Logger default (#350) * weights go into default logger folder * weights go into default logger folder * weights go into default logger folder * weights go into default logger folder * weights go into default logger folder * weights go into default logger folder --- pytorch_lightning/trainer/trainer.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index 7b3b8029..e54ee1ff 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -198,9 +198,10 @@ class Trainer(TrainerIO): # configure checkpoint callback self.checkpoint_callback = checkpoint_callback if self.checkpoint_callback is None: - if isinstance(logger, TestTubeLogger): - ckpt_path = '{}/{}/{}'.format(self.default_save_path, self.logger.name, - self.logger.version) + if isinstance(self.logger, TestTubeLogger): + ckpt_path = '{}/{}/{}/{}'.format(self.default_save_path, + self.logger.name, + self.logger.version, 'checkpoints') else: ckpt_path = self.default_save_path