-
Notifications
You must be signed in to change notification settings - Fork 129
Open
Description
if abs(np.min(mae) - result[0]) < 0.01:
checkpoint_best_path = os.path.join(args.checkpoints_dir, 'best_mae.pth')
torch.save({
'model': model_without_ddp.state_dict(),
}, checkpoint_best_path)
Isn't this wrong? Shouldn't the best model be saved like this-
if(result[0])< np.min(mae):
checkpoint_best_path = os.path.join(args.checkpoints_dir, 'best_mae.pth')
torch.save({
'model': model_without_ddp.state_dict(),
}, checkpoint_best_path)
Metadata
Metadata
Assignees
Labels
No labels