use py3 next syntax

ptrblck · ptrblck · commit e9d1f62fe097 · 2022-07-02T00:25:12.000-07:00
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()
