We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e19cbfc commit e9d1f62Copy full SHA for e9d1f62
examples/int8/training/vgg16/main.py
@@ -93,7 +93,7 @@ def main():
93
model = model.cuda()
94
95
data = iter(training_dataloader)
96
- images, _ = data.next()
+ images, _ = next(data)
97
98
writer.add_graph(model, images.cuda())
99
writer.close()
0 commit comments