Skip to content

Commit 6f71a71

Browse files
author
Moin Nadeem
committed
pyright on trainer file
1 parent b6780eb commit 6f71a71

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

composer/trainer/trainer.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -381,15 +381,15 @@ def create_from_hparams(cls, hparams: TrainerHparams) -> Trainer:
381381
# Checkpoint loading hparams
382382
checkpoint_filepath = hparams.load_checkpoint.filepath if hparams.load_checkpoint is not None else None
383383
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
385385
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
387387

388388
# Checkpoint saving hparams
389389
checkpoint_interval_unit = hparams.save_checkpoint.interval_unit \
390390
if hparams.save_checkpoint is not None else None
391391
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"
393393

394394
trainer = cls(
395395
model=model,

0 commit comments

Comments
 (0)