From fcba947adba0c69d5df3730b0f1ba09abf08c1d6 Mon Sep 17 00:00:00 2001 From: Vasiliy Kuznetsov Date: Sat, 22 Oct 2022 09:43:11 -0700 Subject: [PATCH] [cheery-pick]update NS for FX tutorial for PyTorch v1.13 (#2089) Cherry-pick #2089 --- .jenkins/validate_tutorials_built.py | 1 - prototype_source/fx_numeric_suite_tutorial.py | 4 ++-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/.jenkins/validate_tutorials_built.py b/.jenkins/validate_tutorials_built.py index cc01326b44c..92570124a4e 100644 --- a/.jenkins/validate_tutorials_built.py +++ b/.jenkins/validate_tutorials_built.py @@ -50,7 +50,6 @@ "recipes/Captum_Recipe", "hyperparameter_tuning_tutorial", "flask_rest_api_tutorial", - "fx_numeric_suite_tutorial", # remove when https://github.com/pytorch/tutorials/pull/2089 is fixed ] diff --git a/prototype_source/fx_numeric_suite_tutorial.py b/prototype_source/fx_numeric_suite_tutorial.py index ac43ae49e05..922f48ae0fc 100644 --- a/prototype_source/fx_numeric_suite_tutorial.py +++ b/prototype_source/fx_numeric_suite_tutorial.py @@ -84,9 +84,9 @@ def plot(xdata, ydata, xlabel, ylabel, title): # Note: quantization APIs are inplace, so we save a copy of the float model for # later comparison to the quantized model. This is done throughout the # tutorial. -mobilenetv2_prepared = quantize_fx.prepare_fx( - copy.deepcopy(mobilenetv2_float), qconfig_dict) datum = torch.randn(1, 3, 224, 224) +mobilenetv2_prepared = quantize_fx.prepare_fx( + copy.deepcopy(mobilenetv2_float), qconfig_dict, (datum,)) mobilenetv2_prepared(datum) # Note: there is a long standing issue that we cannot copy.deepcopy a # quantized model. Since quantization APIs are inplace and we need to use