Dataloader

[sezarxray]

HI, What a amazing code!
I need dataloader to run with user-defind steps in each epoch (e.g. 5000 data in train dataset>> 6000 data in each epoch(add duplicate data) ) . Do you have any idea?
thx

[glenn-jocher]

@sezarxray I'm sorry I don't understand the question. The dataloader will use all of the images in your dataset. You can control how many are loaded in each --batch, and you can control the size of your dataset by adding or removing images. See Custom Training Tutorial to get started:

Tutorials

• Train Custom Data  🚀 RECOMMENDED
• Weights & Biases Logging  🌟 NEW
• Multi-GPU Training
• PyTorch Hub  ⭐ NEW
• ONNX and TorchScript Export
• Test-Time Augmentation (TTA)
• Model Ensembling
• Model Pruning/Sparsity
• Hyperparameter Evolution
• Transfer Learning with Frozen Layers  ⭐ NEW
• TensorRT Deployment

[github-actions]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.