File tree Expand file tree Collapse file tree 1 file changed +6
-9
lines changed Expand file tree Collapse file tree 1 file changed +6
-9
lines changed Original file line number Diff line number Diff line change @@ -1607,16 +1607,13 @@ def _load_rng_state(self, checkpoint):
1607
1607
if os .path .isfile (rng_file ):
1608
1608
rng_file_list = paddle .load (rng_file , return_numpy = True )
1609
1609
paddle .distributed .broadcast_object_list (rng_file_list , src = 0 )
1610
- # if rng_file_list still empty, then use old style rng_state
1610
+ # if rng_file_list still empty, not log rng state.
1611
1611
if rng_file_list [0 ] is None :
1612
- rng_file = os .path .join (checkpoint , f"rng_state_{ process_index } .pth" )
1613
- if not os .path .isfile (rng_file ):
1614
- logger .info (
1615
- f"Didn't find an RNG file for process { process_index } , if you are resuming a training that "
1616
- "wasn't launched in a distributed fashion, reproducibility is not guaranteed."
1617
- )
1618
- return
1619
- checkpoint_rng_state = paddle .load (rng_file , return_numpy = True )
1612
+ logger .info (
1613
+ f"Didn't find an RNG file for process { process_index } , if you are resuming a training that "
1614
+ "wasn't launched in a distributed fashion, reproducibility is not guaranteed."
1615
+ )
1616
+ return
1620
1617
else :
1621
1618
checkpoint_rng_state = rng_file_list [process_index ]
1622
1619
else :
You can’t perform that action at this time.
0 commit comments