↓ 1 callersFunctionget_train_sampler(dataset, rank, world_size, global_batch_size, max_steps,
resume_step, seed)
Large-DiT-ImageNet/train.py:53
↓ 1 callersFunctionget_train_sampler(dataset, rank, world_size, global_batch_size, max_steps,
resume_step, seed)
Large-DiT-T2I/train.py:97