Pad experiment version with zero for easier listing (#355)

This commit is contained in:
Yasser Souri
2019-10-10 19:39:26 -04:00
committed by William Falcon
parent 426bb19846
commit 792ba59b78
+1 -1
View File
@@ -260,7 +260,7 @@ class Trainer(TrainerIO):
if self.checkpoint_callback is None:
# init a default one
if isinstance(self.logger, TestTubeLogger):
ckpt_path = '{}/{}/version_{}/{}'.format(
ckpt_path = '{}/{}/version_{:04d}/{}'.format(
self.default_save_path,
self.logger.experiment.name,
self.logger.experiment.version,