From 7440f8b5120bf62efe86274346c4d3da52d1d15c Mon Sep 17 00:00:00 2001 From: GrantPerkins Date: Wed, 1 Nov 2023 13:32:08 -0400 Subject: [PATCH] fix: use correct args for load_checkpoint --- finetune.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/finetune.py b/finetune.py index f0ba943..9fd2c08 100644 --- a/finetune.py +++ b/finetune.py @@ -689,7 +689,7 @@ def main(): best_epoch = None if args.eval_checkpoint: # evaluate the model - load_checkpoint(model, args.eval_checkpoint, args.model_ema) + load_checkpoint(args, model, args.eval_checkpoint, args.model_ema) val_metrics = validate(model, loader_eval, validate_loss_fn, args) print(f"Top-1 accuracy of the model is: {val_metrics['top1']:.1f}%") return