82d414fb55933e83554ad62ab27b164d905860e8,dl/runner.py,AbstractModelRunner,prepare_callbacks,#,351

Before Change

        :return: OrderedDict with callbacks
        
        assert len(kwargs) == 0
        raise NotImplementedError


class ClassificationRunner(AbstractModelRunner):

After Change

        :param **kwargs: callbacks params
        :return: OrderedDict with callbacks
        
        callbacks = OrderedDict()

        for key, value in kwargs.items():
            callback = UtilsFactory.create_callback(**value)
            // @TODO: remove hack
            if isinstance(callback, CheckpointCallback) and resume is not None:
                callback.resume = resume
            if isinstance(callback, InferCallback) and out_prefix is not None:
                callback.out_prefix = out_prefix
            callbacks[key] = callback

        return callbacks


class ClassificationRunner(AbstractModelRunner):

Instances