accelerate
Get rid of calling get_scale() by patching the step method of optimizer.
#1720
Merged
Merged
Get rid of calling get_scale() by patching the step method of optimizer. #1720
sgugger
merged
5 commits into
huggingface:main
from
yuxinyuan:remove_get_scale
Get rid of calling get_scale() by patching the step method of optimizer.
91b9210e
Fix when step() is already patched by other parties.
6dbcb560
support pickle
47e154b6
sgugger
approved these changes
on 2023-07-13
muellerzr
approved these changes
on 2023-07-13
Minor updates.
85f6f4a5
muellerzr
approved these changes
on 2023-07-13
Change _accelerate_num_step_called to _accelerate_step_called
5432d494
sgugger
approved these changes
on 2023-07-14
Login to write a write a comment.
Login via GitHub