def after_training_exp(self, strategy, **kwargs):
        """
        The training method expects an Experience object
        with a 'dataset' parameter.
        """
        for plugin in strategy.plugins:
            if type(plugin) is GenerativeReplayPlugin:
                plugin.generator_strategy.train(strategy.experience)

        The training method expects an Experience object
        with a 'dataset' parameter.
        
avalanche/training/plugins/generative_replay.py