@override
    def create_train_task(self, **kwargs):
        return CaffeTrainTask(framework_id=self.framework_id, **kwargs)
create train task
dataset/ETHPy150Open NVIDIA/DIGITS/digits/frameworks/caffe_framework.py/CaffeFramework.create_train_task