onnxruntime
save_checkpoint, load_checkpoint and aggregate_checkpoints
#6136
Merged
Merged
save_checkpoint, load_checkpoint and aggregate_checkpoints #6136
baijumeswani
merged
12 commits into
master
from
bmeswani/checkpoint
ashbhandare
commented
on 2020-12-16
remove binascii dependence
10e8bb6f
use default values without relying on custom defaults, pass only the …
9e1c1e37
use optimizer states
d296dbfd
save_checkpoint and load_checkpoint implementations
c8cf5269
checkpoint aggregation logic
71e5a6ae
unit tests for save_checkpoint, load_checkpoint and aggregate_checkpo…
42926ea1
optimizer name check, include_optimizer_states for save_checkpoint
b1f8df82
change name from fp32 to full_precision
d196426c
function name change for zero helpers, docstring typo fix
cb26971f
ashbhandare
commented
on 2020-12-17
ashbhandare
commented
on 2020-12-17
ashbhandare
commented
on 2020-12-17
remove redundant assert, helper function for string literals
4bce5f2a
ashbhandare
commented
on 2020-12-17
baijumeswani
commented
on 2020-12-17
load index 0 non aggregation checkpoint file, add trainer options val…
7b5307c0
ashbhandare
dismissed these changes
on 2020-12-17
add comments for using byte string .decode function
583bc201
thiagocrepaldi
approved these changes
on 2020-12-18
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub