Add ability to save optimizer and resume while training#275

coolkp merged 1 commit intomainAI-Hypercomputer/maxdiffusion:mainfrom

optimizer-resumeAI-Hypercomputer/maxdiffusion:optimizer-resumeCopy head branch name to clipboard