in src/data_manager.py [0:0]
def init_data(
dataset_name,
transform,
init_transform,
u_batch_size,
s_batch_size,
classes_per_batch,
unique_classes=False,
multicrop_transform=(0, None),
supervised_views=1,
world_size=1,
rank=0,
root_path=None,
image_folder=None,
training=True,
copy_data=False,
stratify=False,
drop_last=True