riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

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

Browse Source 
* no conv bn folding in ipex

* no flag in training

* comment

Co-authored-by: Sander Land <sander@chatdesk.com>
This commit is contained in:
Sander Land 2022-10-26 14:58:52 +02:00 committed by GitHub
parent 802b98c72b
commit fdffee8a60
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/trainer.py
View File

@ -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()