2021SC@SDUSC
本篇代码分析模块为:train.py
一、引入包简介
from Load_dataset import load_split_MUTAG_data, accuracy
from Model import GCN
import time
import numpy as np
import torch
import torch.optim as optim
import torch.nn.functional as F
torch.optim:
功能:优化更新深度学习模型的参数
介绍:实现了深度学习中绝大多数的优化方法,例如RMSProp,Adam,SGD等,更便于应用,所以通常不需要手动写上述代码。
使用:要使用 torch.optim
要先创建一个optim 对象,这个对象会一直保持当前状态或根据计算的梯度更新参数。创建optim对象时,要给它一个包含模型参数的的可迭代对象(所有的都应该是 Variable ),然后指定learning rate,weight decay等参数。
二、EarlyStopping类分析
这个类其实涉及到了关于