diff --git a/evaluate_auc.py b/evaluate_auc.py index 932eef3..5b6e605 100644 --- a/evaluate_auc.py +++ b/evaluate_auc.py @@ -452,7 +452,6 @@ def main(): # Load model checkpoint and initialize model. model = load_model( config_path=f'config_{model_name}.json', - model_path=f'best_model_{model_name}.pt', device=device, ) model.eval()