fixes Flake8

This commit is contained in:
William Falcon 2019-10-09 17:49:29 -04:00
parent 0eab1e42b2
commit 96c2a2de50
1 changed files with 5 additions and 3 deletions

View File

@ -260,9 +260,11 @@ class Trainer(TrainerIO):
if self.checkpoint_callback is None:
# init a default one
if isinstance(self.logger, TestTubeLogger):
ckpt_path = '{}/{}/version_{}/{}'.format(self.default_save_path,
self.logger.experiment.name,
self.logger.experiment.version, 'checkpoints')
ckpt_path = '{}/{}/version_{}/{}'.format(
self.default_save_path,
self.logger.experiment.name,
self.logger.experiment.version,
'checkpoints')
else:
ckpt_path = self.default_save_path