Pj12 commited on
Commit
0cf565d
·
verified ·
1 Parent(s): b222927

Update train_nsf_sim_cache_sid_load_pretrain.py

Browse files
train_nsf_sim_cache_sid_load_pretrain.py CHANGED
@@ -319,8 +319,8 @@ def run(rank, n_gpus, hps):
319
  if epoch-1 == epoch_str:
320
  elapsed_time_record.pop(0)
321
  elapsed_time_avg = sum(elapsed_time_record)/len(elapsed_time_record)
322
- time_left = elapsed_time_avg*(hps.total_epoch-epoch)
323
- time_ovt = elapsed_time_avg*(overtrain-epoch)
324
  hour, minute, second, millisec = Calculate_format_elapsed_time(elapsed_time)
325
  hour_left, minute_left, second_left, millisec_left = Calculate_format_elapsed_time(time_left)
326
  hour_ovt, minute_ovt, second_ovt, millisec_ovt = Calculate_format_elapsed_time(time_ovt)
 
319
  if epoch-1 == epoch_str:
320
  elapsed_time_record.pop(0)
321
  elapsed_time_avg = sum(elapsed_time_record)/len(elapsed_time_record)
322
+ time_left = elapsed_time_avg*(hps.total_epoch - epoch)
323
+ time_ovt = elapsed_time_avg*(overtrain - (len(Loss_Gen_Per_Epoch) - right_index(Loss_Gen_Per_Epoch,min(Loss_Gen_Per_Epoch)) + 1)) # 5/1/2025
324
  hour, minute, second, millisec = Calculate_format_elapsed_time(elapsed_time)
325
  hour_left, minute_left, second_left, millisec_left = Calculate_format_elapsed_time(time_left)
326
  hour_ovt, minute_ovt, second_ovt, millisec_ovt = Calculate_format_elapsed_time(time_ovt)