Fix silly train.py typo during dataset work

pull/323/head
Ross Wightman 4 years ago
parent d55bcc0fee
commit 9d5d4b8df6

@ -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,

Loading…
Cancel
Save