Pytorch 加载 模型

Reusable GitLab CI/CD Component for MLOps: data validation, model training with MLflow, quality gates, and model registration. Supports sklearn, PyTorch, TensorFlow, XGBoost, LightGBM.

1. 方式一: 加载和保存状态字典 只保存参数的方法在加载的时候要事先定义好跟原模型一致的模型，并在该模型的实例对象(假设名为model)上进行加载，即在使用上述加载语句前已经有定义了一个和原模型一样的Net, 并且进行了实例化 model=Net( ) . ## 保存模型的字典参数， 保存模型的后