Update docs/tutorials/pretrained_weights.ipynb

microsoft · Dec 18, 2023 · 31709ac · 31709ac
1 parent 1e19d7d
commit 31709ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/tutorials/pretrained_weights.ipynb b/docs/tutorials/pretrained_weights.ipynb
@@ -245,7 +245,7 @@
    "outputs": [],
    "source": [
     "in_chans = weights.meta[\"in_chans\"]\n",
-    "model = timm.create_model(\"resnet18\", in_chans=in_chans, num_classes=256)\n",
+    "model = timm.create_model(\"resnet18\", in_chans=in_chans, num_classes=10)\n",
     "model.load_state_dict(weights.get_state_dict(progress=True), strict=False)"
    ]
   },