message: chore: Fix dataloader in finetune_qat script
diff: @@ -261,7 +261,7 @@ def main():
  state = ckpt["state"]
  data = iter(training_dataloader)
  - images, _ = data.next()
  + images, _ = next(data)
  writer.add_graph(model, images.cuda())
  writer.close()
  