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: ...@@ -1280,7 +1280,8 @@ class Trainer:
if not training: if not training:
model.eval() 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: else:
if not model.training: if not model.training:
model.train() 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