assert params.start_epoch>0, params.start_epoch

This commit is contained in:
yaozengwei 2022-05-07 10:36:04 +08:00
parent 0bee5d058a
commit 537617302a

View File

@ -861,7 +861,7 @@ def run(rank, world_size, args):
# model_avg is only used with rank 0
model_avg = copy.deepcopy(model)
assert params.start_epoch > 0
assert params.start_epoch > 0, params.start_epoch
checkpoints = load_checkpoint_if_available(
params=params, model=model, model_avg=model_avg
)