SuYee Liew
Results
2
comments of
SuYee Liew
same issue here, got the error `_pickle.PicklingError: Cannot pickle a prepared model with automatic mixed precision, please unwrap the model with `Accelerator.unwrap_model(model)` before pickling it.`
setting `save_model_class=False` worked for me. Just need to load the model class manually.