transformers
Extend Transformers Trainer Class to Enable PyTorch Torchscript for Inference
#17153
Merged
Merged
Extend Transformers Trainer Class to Enable PyTorch Torchscript for Inference #17153
sgugger
merged
28 commits into
huggingface:main
from
jianan-gu:Introduce_Jit
add jit mode option and model wrap
4572bd61
Merge branch 'main' into Introduce_Jit
ff22924f
Update src/transformers/training_args.py
b7d87d4e
Update src/transformers/training_args.py
7be2f3f7
Merge branch 'huggingface:main' into Introduce_Jit
74a806ec
refine code
8ab4ad31
Update src/transformers/trainer.py
fc197cc3
Update src/transformers/trainer.py
1659d18a
add ut and refine code
e8bd0114
Merge branch 'main' into Introduce_Jit
f516fdba
code refine
e1727adc
refine code
2d7b0a85
add inference doc
11ffcda2
Update src/transformers/trainer.py
8a07c6f9
Update src/transformers/trainer.py
c5ff4ae9
add cpu inference performance doc
5297c702
Update perf_infer_cpu.mdx
1351b970
stas00
commented
on 2022-06-10
Update perf_infer_cpu.mdx
973375fa
Update performance.mdx
5f60fe79
Update _toctree.yml
aff4ac5d
refine jit func naming
e96d9cf2
Update _toctree.yml
ce8b230b
Delete perf_infer_gpu_one.mdx
04a36bd7
Update perf_infer_cpu.mdx
216b289e
Update docs/source/en/perf_infer_cpu.mdx
54cd5591
add none check before jit
ae377f59
stas00
approved these changes
on 2022-06-14
Update docs/source/en/perf_infer_cpu.mdx
3d78cb58
Update docs/source/en/perf_infer_cpu.mdx
ffdea683
Login to write a write a comment.
Login via GitHub