chore: Fix dataloader in finetune_qat script

Signed-off-by: Anurag Dixit <[email protected]>

Author: andi4191

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