from local

This commit is contained in:
dohe0342 2023-02-03 01:30:45 +09:00
parent a6dcaff260
commit bdc439c678
2 changed files with 1 additions and 0 deletions

View File

@ -475,6 +475,7 @@ def compute_validation_loss(
for k, v in params.best_valid_losses.items():
if v == max(params.best_valid_losses.values()):
remove_epoch = k
break
remove_filename = params.exp_dir / f"best-valid-loss_{remove_epoch}.pt"
try: os.remove(remove_filename)