File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -2946,7 +2946,7 @@ def scale_and_round(x):
29462946 ext_separated_list_of_batches[x] = holder[:]
29472947 '''
29482948 distributed_state .wait_for_everyone ()
2949- '''
2949+
29502950 if distributed_state .is_main_process :
29512951 batchlogstr = "Running through ext_separated_list_of_batches Before Gather:\n "
29522952 for x in range (len (ext_separated_list_of_batches )):
@@ -2956,7 +2956,7 @@ def scale_and_round(x):
29562956 for z in range (len (ext_separated_list_of_batches [x ][y ])):
29572957 batchlogstr += f" Image { z } of { len (ext_separated_list_of_batches [x ][y ])} break: { ext_separated_list_of_batches [x ][y ][z ].global_count } \n "
29582958 logger .info (batchlogstr )
2959- '''
2959+
29602960 ext_separated_list_of_batches = gather_object (ext_separated_list_of_batches )
29612961 del extinfo
29622962 #logger.info(f"\nDevice {distributed_state.device}: {ext_separated_list_of_batches}")
You can’t perform that action at this time.
0 commit comments