diff --git a/notebooks/image-classifier.ipynb b/notebooks/image-classifier.ipynb index 33f506696b..921e32e61d 100644 --- a/notebooks/image-classifier.ipynb +++ b/notebooks/image-classifier.ipynb @@ -28,7 +28,7 @@ }, { "cell_type": "code", - "execution_count": 2, + "execution_count": null, "metadata": {}, "outputs": [ { @@ -74,7 +74,7 @@ "\n", "# Grab images from our training data\n", "dataiter = iter(trainloader)\n", - "images, labels = dataiter.next()\n", + "images, labels = next(dataiter)\n", "\n", "for i in range(batch_size):\n", " # Add new subplot\n", @@ -208,7 +208,7 @@ }, { "cell_type": "code", - "execution_count": 5, + "execution_count": null, "metadata": {}, "outputs": [ { @@ -226,7 +226,7 @@ "# Pick random photos from training set\n", "if dataiter == None:\n", " dataiter = iter(testloader)\n", - "images, labels = dataiter.next()\n", + "images, labels = next(dataiter)\n", "\n", "# Load our model\n", "net = Net()\n", diff --git a/requirements.txt b/requirements.txt index 0b99362c31..c40ef94b81 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,8 +1,8 @@ ipywidgets==8.1.8 matplotlib==3.8.4 pandas==2.2.2 -torch==2.10.0 -torchvision==0.25.0 +torch==2.13.0 +torchvision==0.28.0 tqdm==4.66.4 pillow>=12.1.1 fonttools>=4.60.0