如何在PyTorch里面使用神经网络
在PyTorch里面写神经网络,所有的层结构和损失函数都来自于torch.nn,所有的模型构建都是从nn.Module这个基类继承的。
模型的创建过程就是定义一个class继承nn.Module,然后重写init方法和forward方法,实例代码如下:
import torch.nn as nn
import torch.nn.functional as F
class Model(nn.Module):
def __init__(self):
super(Model, se
原创
2021-04-26 08:57:47 ·
182 阅读 ·
0 评论