diff --git a/.jenkins/validate_tutorials_built.py b/.jenkins/validate_tutorials_built.py index 5f9d563475..b426049b58 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 "ax_multiobjective_nas_tutorial", ] diff --git a/prototype_source/fx_numeric_suite_tutorial.py b/prototype_source/fx_numeric_suite_tutorial.py index ac43ae49e0..922f48ae0f 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