Skip to content

Commit

Permalink
Dist ckpt refactor (#8)
Browse files Browse the repository at this point in the history
* refactored for distributed checkpointing

* addressing comments

* missed commit
  • Loading branch information
adil-a authored Apr 8, 2024
1 parent 1e15d64 commit 8320c48
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions vectorlm/utils/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,12 +12,10 @@
checkpoint_exists,
get_latest_checkpoint_dir,
load_metadata,
load_model,
load_optimizer,
load_model_and_optimizer,
load_scheduler,
save_consolidated_model,
save_metadata,
save_model,
save_optimizer,
save_model_and_optimizer,
save_scheduler,
)

0 comments on commit 8320c48

Please sign in to comment.