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.
2 parents 4694c69 + e9d1f62 commit 45a57daCopy full SHA for 45a57da
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