search

victoresque / pytorch-template

PyTorch deep learning projects made easy.
MIT License
4.7k stars 1.08k forks source link

DataLoader example #110

Open fneaplle opened 1 year ago

fneaplle commented 1 year ago

I think DataLoader Usage example need to be changed.(enumerate should be added)

-----Before------- DataLoader Usage

BaseDataLoader is an iterator, to iterate through batches:

for batch_idx, (x_batch, y_batch) in data_loader: pass

------After--------- DataLoader Usage

BaseDataLoader is an iterator, to iterate through batches:

for batch_idx, (x_batch, y_batch) in enumerate(data_loader): pass