diff --git a/tools/pytorch-quantization/examples/calibrate_quant_resnet50.ipynb b/tools/pytorch-quantization/examples/calibrate_quant_resnet50.ipynb index bf0c427a..17e4377e 100644 --- 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",