message: use py3 next syntax
diff: @@ -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()
  