Error when saving model in accelerate

[muellerzr]

@aclifton314

single model or do I need to indicate that accelerate should wait until all the processes are done calculating?

It will save a single model. get_state_dict will perform all of those bits for you so you don’t have to worry about that