From 9d5d4b8df631a826ce5f1b56e6e9c8b3c5ea1973 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sat, 16 Jan 2021 16:32:21 -0800 Subject: [PATCH] Fix silly train.py typo during dataset work --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index b31199f9..a4010e1f 100755 --- a/train.py +++ b/train.py @@ -551,7 +551,7 @@ def main(): try: for epoch in range(start_epoch, num_epochs): 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( epoch, model, loader_train, optimizer, train_loss_fn, args,