File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -381,15 +381,15 @@ def create_from_hparams(cls, hparams: TrainerHparams) -> Trainer:
381
381
# Checkpoint loading hparams
382
382
checkpoint_filepath = hparams .load_checkpoint .filepath if hparams .load_checkpoint is not None else None
383
383
checkpoint_load_weights_only = hparams .load_checkpoint .load_weights_only \
384
- if hparams .load_checkpoint is not None else None
384
+ if hparams .load_checkpoint is not None else False
385
385
checkpoint_strict_model_weights = hparams .load_checkpoint .strict_model_weights \
386
- if hparams .load_checkpoint is not None else None
386
+ if hparams .load_checkpoint is not None else False
387
387
388
388
# Checkpoint saving hparams
389
389
checkpoint_interval_unit = hparams .save_checkpoint .interval_unit \
390
390
if hparams .save_checkpoint is not None else None
391
391
checkpoint_interval = hparams .save_checkpoint .interval if hparams .save_checkpoint is not None else None
392
- checkpoint_folder = hparams .save_checkpoint .folder if hparams .save_checkpoint is not None else None
392
+ checkpoint_folder = hparams .save_checkpoint .folder if hparams .save_checkpoint is not None else "checkpoints"
393
393
394
394
trainer = cls (
395
395
model = model ,
You can’t perform that action at this time.
0 commit comments