mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix silly train.py typo during dataset work

Browse Source
This commit is contained in:
Ross Wightman 2021-01-16 16:32:21 -08:00
parent d55bcc0fee
commit 9d5d4b8df6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -551,7 +551,7 @@ def main():
try: try:
for epoch in range(start_epoch, num_epochs): for epoch in range(start_epoch, num_epochs):
if args.distributed and hasattr(loader_train.sampler, 'set_epoch'): if args.distributed and hasattr(loader_train.sampler, 'set_epoch'):
loader_train.set_epoch(epoch) loader_train.sampler.set_epoch(epoch)
train_metrics = train_one_epoch( train_metrics = train_one_epoch(
epoch, model, loader_train, optimizer, train_loss_fn, args, epoch, model, loader_train, optimizer, train_loss_fn, args,