From 149033c529749f7be5bb3368fa694f2267c54a91 Mon Sep 17 00:00:00 2001 From: Anurag Dixit Date: Thu, 18 Aug 2022 19:03:37 -0700 Subject: [PATCH] chore: Fix dataloader in finetune_qat script Signed-off-by: Anurag Dixit --- examples/int8/training/vgg16/finetune_qat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/int8/training/vgg16/finetune_qat.py b/examples/int8/training/vgg16/finetune_qat.py index 48709d9f8a..6ec20a9a46 100644 --- a/examples/int8/training/vgg16/finetune_qat.py +++ b/examples/int8/training/vgg16/finetune_qat.py @@ -261,7 +261,7 @@ def main(): state = ckpt["state"] data = iter(training_dataloader) - images, _ = data.next() + images, _ = next(data) writer.add_graph(model, images.cuda()) writer.close()