accelerate
Add kwargs to optimizer, scheduler and dataloader using function `accelerator().load_state()`
#3540
Merged
Merged
Add kwargs to optimizer, scheduler and dataloader using function `accelerator().load_state()` #3540
SunMarc
merged
18 commits into
huggingface:main
from
luiz0992:luiz/add-kwargs-to-load-state
Added artifacts and figure tracking at MLFlow tracker
d42384dd
Added `log_artifact` to the MLFlowTracker
5d051b9c
Remove changes
748a5877
Merge branch 'huggingface:main' into main
30a31456
Added kwargs when loading state.
b903eef9
added doc string
7f54f124
luiz0992
changed the title
luiz0992
changed the title
Adjusted correct default types of kwargs
e07f9ff5
SunMarc
commented
on 2025-05-12
Changed the load kwargs to a single one
810c69ee
removed None value from kwargs
e3c4327d
Merge branch 'huggingface:main' into luiz/add-kwargs-to-load-state
0dd2077e
fix kwargs for loading the model
3cfa4c2d
removed load_kwargs from optimizer state dict
c96107fa
make load_kwargs a dictionary
13e6e6e2
revert last changes
17beba70
SunMarc
commented
on 2025-05-15
reverted load_kwargs
c3d7f1fe
fix docstring
4381b93b
SunMarc
approved these changes
on 2025-05-19
SunMarc
commented
on 2025-05-19
added dict initiation
e54694ce
Fix quality error during PR
9980afb9
Login to write a write a comment.
Login via GitHub