Open FacerAin opened 1 year ago
Model save error when including lr_scheduler like this Can‘t pickle local object‘get_linear_schedule_with_warmup’\<locals\> lr_lamda‘
Can‘t pickle local object‘get_linear_schedule_with_warmup’\<locals\> lr_lamda‘
You can solve this error to add dill library like this
import dill ... torch.save(model, "model.pt", pickle_module=dill)
referenced by: https://blog.csdn.net/qq_37747189/article/details/117700335
Model save error when including lr_scheduler like this
Can‘t pickle local object‘get_linear_schedule_with_warmup’\<locals\> lr_lamda‘
You can solve this error to add dill library like this
referenced by: https://blog.csdn.net/qq_37747189/article/details/117700335