enable user to pass in save_filename

mosaicml / examples

Fast and flexible reference benchmarks

Apache License 2.0

441 stars 125 forks source link

Closed vchiley closed 1 year ago

vchiley commented 1 year ago

using save_filename: ep{epoch}-ba{batch}/rank{rank}.pt puts sharded ckpts in folder structure.