参考:pytorch教程之nn.Module类详解——使用Module类来自定义模型
一、简单的模型模板
1、定义网络结构
class MyNet(nn.Module):
# 初始化函数 __init__(self):
# 定义了具体网络有那些层,但并没有决定网络的结构。
def __init__(self) -> None:
super
参考:pytorch教程之nn.Module类详解——使用Module类来自定义模型
一、简单的模型模板
1、定义网络结构
class MyNet(nn.Module):
# 初始化函数 __init__(self):
# 定义了具体网络有那些层,但并没有决定网络的结构。
def __init__(self) -> None:
super