diff --git a/16_tensorboard.py b/16_tensorboard.py index 0bd67dc6..ba2045cd 100644 --- a/16_tensorboard.py +++ b/16_tensorboard.py @@ -43,7 +43,7 @@ shuffle=False) examples = iter(test_loader) -example_data, example_targets = examples.next() +example_data, example_targets = next(examples) # previously `examples.next()` for i in range(6): plt.subplot(2,3,i+1) @@ -156,4 +156,4 @@ def forward(self, x): preds_i = class_preds[:, i] writer.add_pr_curve(str(i), labels_i, preds_i, global_step=0) writer.close() - ################################################### \ No newline at end of file + ###################################################