Method _method_callback

python/oneflow/framework/model.py:353–356

(self, method_name: str = None, *args, **kwargs)

        ), "model callbacks' type must be model.Callback or List[model.Callback]."

    def _method_callback(self, method_name: str = None, *args, **kwargs):
        for cb in self._cbs:
            method = getattr(cb, method_name)
            method(*args, **kwargs)


class TrainModel(SubModel):