torchvision notebook error fix (#45)

bfineran · web-flow · commit 3e874ad4a661 · 2021-02-01T15:57:12.000-05:00
diff --git a/examples/pytorch-torchvision/pruning.ipynb b/examples/pytorch-torchvision/pruning.ipynb
@@ -98,7 +98,7 @@
     "# Overriding number of classes\n",
     "NUM_CLASSES = 10  # number of imagenette classes\n",
     "model.fc = Linear(in_features=model.fc.in_features, out_features=NUM_CLASSES, bias=True)\n",
-    "print(fc)"
+    "print(model.fc)"
    ]
   },
   {

Original file line number	Diff line number	Diff line change
`@@ -98,7 +98,7 @@`
`98`	`98`	`"# Overriding number of classes\n",`
`99`	`99`	`"NUM_CLASSES = 10 # number of imagenette classes\n",`
`100`	`100`	`"model.fc = Linear(in_features=model.fc.in_features, out_features=NUM_CLASSES, bias=True)\n",`
`101`		`- "print(fc)"`
	`101`	`+ "print(model.fc)"`
`102`	`102`	`]`
`103`	`103`	`},`
`104`	`104`	`{`