evaluate_auc: adapt to new utils.load_model signature (infer checkpoint from config)
This commit is contained in:
@@ -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()
|
||||
|
Reference in New Issue
Block a user