Unverified Commit fdffee8a authored by Sander Land's avatar Sander Land Committed by GitHub
Browse files

No conv bn folding in ipex to avoid warning (#19870)



* no conv bn folding in ipex

* no flag in training

* comment
Co-authored-by: default avatarSander Land <sander@chatdesk.com>
parent 802b98c7
......@@ -1280,7 +1280,8 @@ class Trainer:
if not training:
model.eval()
model = ipex.optimize(model, dtype=dtype, level="O1")
# conv_bn_folding is disabled as it fails in symbolic tracing, resulting in ipex warnings
model = ipex.optimize(model, dtype=dtype, level="O1", conv_bn_folding=False)
else:
if not model.training:
model.train()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment