mlfoundations / open_lm

A repository for research on medium sized language models.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Factorize helper function for all model loading

sagadre opened this issue · comments

It would be nice to simplify the checkpoint loading. Right now it is a bit confusing with checkpoint_path, args.checkpoint_path, join(.., "checkpoints"), remote-sync, etc. all referenced in various places inmain.py