ENVs
DNN
CNN
RNN
GEN
GNN
NLP
Network
GPU/TPU
TensorFlow
PyTorch
utils
import torch
# 保存模型 torch.save(the_model.state_dict(), PATH) # 加载模型 the_model = TheModelClass(*args, **kwargs) the_model.load_state_dict(torch.load(PATH))
# 保存模型 torch.save(the_model, PATH) # 加载模型 the_model = torch.load(PATH)