Pytorch 加载 模型

A clean, modular implementation of Physics-Informed Neural Networks (PINNs) for solving the 1D and 2D wave equation using PyTorch.

CUDA + PyTorch

BacktoPyTorch Index 模型保存 # 保存整个模型的结构信息和参数信息, 保存的对象是模型 model ，可以是pth方式，也可以是pkl的方式torch.save(model,'./model.pth')# 方式1# 保存模型的参数, 保存的对象是模型的状态 model.state