1f2a21411601c92234bc4d11d59eb5a577fe1ecd,catalyst/core/runner.py,_Runner,_prepare_for_stage,#_Runner#Any#,154

Before Change

            self._get_experiment_components(stage)

        utils.set_global_seed(self.experiment.initial_seed)
        self.state = self.state_fn(
            stage=stage,
            model=self.model,
            device=self.device,
            criterion=criterion,
            optimizer=optimizer,
            scheduler=scheduler,
            **self.experiment.get_state_params(stage),
            **migrating_params
        )

After Change

    def _prepare_for_stage(self, stage: str):
        utils.set_global_seed(self.experiment.initial_seed)
        migrating_params = {}
        stage_state_params = self.experiment.get_state_params(stage)
        migrate_from_previous_stage = \
            stage_state_params.get("migrate_from_previous_stage", True)
        if self.state is not None and migrate_from_previous_stage:
            migrating_params.update(
                {
                    "step": self.state.step,
                    "epoch": self.state.epoch
                }
            )

        utils.set_global_seed(self.experiment.initial_seed)
        self.model, criterion, optimizer, scheduler, self.device = \
            self._get_experiment_components(stage)

        utils.set_global_seed(self.experiment.initial_seed)
        self.state = self.state_fn(
            stage=stage,
            model=self.model,
            device=self.device,

Instances