Merge pull request #1292 from pytorch/anuragd/fix_finetune_qat_dataloader

chore: Fix dataloader in finetune_qat script

Author: peri044

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()