Pj12 commited on
Commit
2dc975e
·
verified ·
1 Parent(s): 1292e78

Update train_nsf_sim_cache_sid_load_pretrain.py

Browse files
train_nsf_sim_cache_sid_load_pretrain.py CHANGED
@@ -289,7 +289,7 @@ def run(rank, n_gpus, hps):
289
  #
290
  Lowest_lg = f"{min(Loss_Gen_Per_Epoch):.5f}, epoch: {right_index(Loss_Gen_Per_Epoch,min(Loss_Gen_Per_Epoch))+1}"
291
  Lowest_ld = f"{min(Loss_Disc_Per_Epoch):.5f}, epoch: {right_index(Loss_Disc_Per_Epoch,min(Loss_Disc_Per_Epoch))+1}"
292
- print(f"{hps.name}_e{epoch}_s{global_step} | Loss gen total: {Loss_Gen_Per_Epoch[-1]:.5f} | Lowest loss G: {Lowest_lg}\n Loss disc: {Loss_Disc_Per_Epoch[-1]:.5f} | Lowest loss D: {Lowest_ld}")
293
  print(f"Specific Value: loss gen={loss_gen:.3f}, loss fm={loss_fm:.3f}, loss mel={loss_mel:.3f}, loss kl={loss_kl:.3f}")
294
  #
295
  if len(Loss_Gen_Per_Epoch) > Min_for_Single_epoch and epoch % hps.save_every_epoch != 0:
 
289
  #
290
  Lowest_lg = f"{min(Loss_Gen_Per_Epoch):.5f}, epoch: {right_index(Loss_Gen_Per_Epoch,min(Loss_Gen_Per_Epoch))+1}"
291
  Lowest_ld = f"{min(Loss_Disc_Per_Epoch):.5f}, epoch: {right_index(Loss_Disc_Per_Epoch,min(Loss_Disc_Per_Epoch))+1}"
292
+ print(f"{hps.name}_e{epoch}_s{global_step} | Loss gen total: {Loss_Gen_Per_Epoch[-1]:.8f} | Lowest loss G: {Lowest_lg}\n Loss disc: {Loss_Disc_Per_Epoch[-1]:.8f} | Lowest loss D: {Lowest_ld}")
293
  print(f"Specific Value: loss gen={loss_gen:.3f}, loss fm={loss_fm:.3f}, loss mel={loss_mel:.3f}, loss kl={loss_kl:.3f}")
294
  #
295
  if len(Loss_Gen_Per_Epoch) > Min_for_Single_epoch and epoch % hps.save_every_epoch != 0: