Function Parallelize_GPU_BMUF

caffe2/python/data_parallel_model.py:417–419

(*args, **kwargs)

def Parallelize_GPU_BMUF(*args, **kwargs):
    kwargs['cpu_device'] = False
    Parallelize_BMUF(*args, **kwargs)


def Parallelize_CPU_BMUF(*args, **kwargs):