Merge pull request #1159 from ptrblck/next_fix

narendasan · web-flow · commit 45a57da8f715 · 2022-07-05T12:45:28.000-06:00
Use py3 next() syntax
diff --git a/examples/int8/training/vgg16/main.py b/examples/int8/training/vgg16/main.py
@@ -93,7 +93,7 @@ def main():
     model = model.cuda()
 
     data = iter(training_dataloader)
-    images, _ = data.next()
+    images, _ = next(data)
 
     writer.add_graph(model, images.cuda())
     writer.close()
