We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ede1b87 commit 986c958Copy full SHA for 986c958
python/paddle/distributed/fleet/meta_parallel/parallel_layers/pp_layers.py
@@ -674,9 +674,6 @@ def _build_layer(self):
674
paddle.set_rng_state(orig_rng_state)
675
get_rng_state_tracker().set_states_tracker(orig_rng_tracker)
676
677
- paddle.set_rng_state(orig_rng_state)
678
- get_rng_state_tracker().set_states_tracker(orig_rng_tracker)
679
-
680
def _build_layer_impl(self, start, end):
681
if self._num_virtual_pipeline_stages > 1:
682
# For interleave scheduler, all layers relating with one model chunk will be saved in PipelineLayerChunk
0 commit comments