INNER CODE UNIT · Python

create

KaiyangZhou/pytorch-center-loss · datasets.py:38

def create(name, batch_size, use_gpu, num_workers):
    if name not in __factory.keys():
        raise KeyError("Unknown dataset: {}".format(name))
    return __factory[name](batch_size, use_gpu, num_workers)

View source record →