def before_training(self, strategy: BaseSGDTemplate, *args, **kwargs):
        """Called before `train` by the `BaseTemplate`."""

        # Save model's initial weights in the first experience training step
        if self.initial_weights is None:
            # Save initial weights
            self.initial_weights = deepcopy(strategy.model.state_dict())
Called before `train` by the `BaseTemplate`.
avalanche/training/plugins/from_scratch_training.py