1.在定义模型时,def __init__(): 写一些你用到了了那些层
2.前向传播: 在上面写了这么多层,这些层怎么用的,怎么搭配。
3.优化器:optimizer = torch.optim.SGD( 需要优化的参数 eg: model.paramenters(), lr = learning_rate)
调用的时候:optimizer.step()
4.把模型放到GPU上
把数据放到GPU上
5.创建 module 来更简化代码
6.train
1.在定义模型时,def __init__(): 写一些你用到了了那些层
2.前向传播: 在上面写了这么多层,这些层怎么用的,怎么搭配。
3.优化器:optimizer = torch.optim.SGD( 需要优化的参数 eg: model.paramenters(), lr = learning_rate)
调用的时候:optimizer.step()
4.把模型放到GPU上
把数据放到GPU上
5.创建 module 来更简化代码
6.train