Pj12 commited on
Commit
64fd7ff
·
verified ·
1 Parent(s): 5d2f297

Update train_nsf_sim_cache_sid_load_pretrain.py

Browse files
train_nsf_sim_cache_sid_load_pretrain.py CHANGED
@@ -694,6 +694,8 @@ def train_and_evaluate(
694
  ),
695
  )
696
  )
 
 
697
 
698
  try:
699
  with open("csvdb/stop.csv") as CSVStop:
 
694
  ),
695
  )
696
  )
697
+ if min(Loss_Gen_Per_Epoch[Min_for_Single_epoch::1]) == Loss_Gen_Per_Epoch[-1]:
698
+ os.rename(f"logs/{hps.name}/weights/{hps.name}_e{epoch}_s{global_step}.pth",f"logs/{hps.name}/weights/{hps.name}_e{epoch}_s{global_step}_Best_Epoch.pth")
699
 
700
  try:
701
  with open("csvdb/stop.csv") as CSVStop: