pytorch-quantization: fix undefined 'model_name'

Signed-off-by: Rajeev Rao <[email protected]>
wenjunnutter · Sep 22, 2021 · f5ea4e1 · f5ea4e1
1 parent 5325570
commit f5ea4e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/pytorch-quantization/examples/calibrate_quant_resnet50.ipynb b/tools/pytorch-quantization/examples/calibrate_quant_resnet50.ipynb
@@ -529,7 +529,7 @@
     "traindir = os.path.join(data_path, 'train')\n",
     "valdir = os.path.join(data_path, 'val')\n",
     "_args = collections.namedtuple('mock_args', ['model', 'distributed', 'cache_dataset'])\n",
-    "dataset, dataset_test, train_sampler, test_sampler = load_data(traindir, valdir, _args(model=model_name, distributed=False, cache_dataset=False))\n",
+    "dataset, dataset_test, train_sampler, test_sampler = load_data(traindir, valdir, _args(model='resnet50', distributed=False, cache_dataset=False))\n",
     "\n",
     "data_loader = torch.utils.data.DataLoader(\n",
     "    dataset, batch_size=batch_size,\n",