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-
|
| 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)
|