In some models (I found it in resnet50, vgg16, bert_base, roberta, but please check other too) the disable_jit_freeze parameter isn't correctly passed to the run_pytorch_fp function.
It makes tests with jit freeze not working. It is very useful testing the eager mode.
In some models (I found it in resnet50, vgg16, bert_base, roberta, but please check other too) the
disable_jit_freezeparameter isn't correctly passed to therun_pytorch_fpfunction.It makes tests with jit freeze not working. It is very useful testing the eager mode.