wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-28 13:20:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Logger default (#350)

Browse Source 
* 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
This commit is contained in:
William Falcon authored and GitHub committed 2019-10-09 16:25:04 -04:00
1 parent 6e0a562ecb
commit d95e693598
1 file changed
+4 -3
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/trainer.py
+4 -3
View File
@@ -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