|
@@ -103,7 +103,7 @@ class StepControl(MPFuture):
|
|
@stage.setter
|
|
@stage.setter
|
|
def stage(self, stage: AveragingStage):
|
|
def stage(self, stage: AveragingStage):
|
|
if stage == AveragingStage.RUNNING_ALLREDUCE:
|
|
if stage == AveragingStage.RUNNING_ALLREDUCE:
|
|
- self.can_modify = False
|
|
|
|
|
|
+ self.began_allreduce = True
|
|
self._shared_buffer[StepControl._STAGE] = stage.value
|
|
self._shared_buffer[StepControl._STAGE] = stage.value
|
|
|
|
|
|
@property
|
|
@property
|