- 博客(1)
- 收藏
- 关注
原创 如何在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
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人