pytorch的学习之路(一)| 模型的forward方法是如何被调用的
问题out = net(image) # 图像作为输入,经过net做正向传播,得到输出(分类/框/。。。)你有没有一个疑问,上面这行代码是如何调用forward()函数得到结果的?我会贴出源码并做解释解答一步一步跟踪,net(image)到底经历了什么?(以下引用该开源代码做讲解,其中会做适当简化,以达到说明的目的)net的定义net = RetinaFace()RetinaFace类的定义class RetinaFace(nn.Module): def __init__(s
原创
2020-08-21 15:34:45 ·
10261 阅读 ·
8 评论