diff --git a/.jenkins/validate_tutorials_built.py b/.jenkins/validate_tutorials_built.py index 5f9d563475f..b426049b583 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 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 diff --git a/requirements.txt b/requirements.txt index 67bec81cd2b..53ab4553317 100644 --- a/requirements.txt +++ b/requirements.txt @@ -28,6 +28,10 @@ ax-platform nbformat>=4.2.0 deep_phonemizer==0.0.17 +# the following is necessary due to https://github.com/python/importlib_metadata/issues/411 +importlib-metadata < 5.0; python_version <= "3.7" +importlib-metadata; python_version > "3.7" + # PyTorch Theme -e git+https://github.com/pytorch/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme