【深度之眼】Pytorch框架班第五期-模型创建机制

调试 1、以LeNet为例,我们在第72行设置断点,并进行Debug。 2、点击C按钮进入到模型中,我们可以发现__init__()函数构建了LeNet模型需要的子模块。 3、我们一直点击B按钮,当最后一个模块构建完成后,程序跳出,LeNet模型的初始化就完成了。模型构建完成。 4、我们将代码运行到模型训练部分的97行,此处我们对模型输入了inputs,输出outputs。 5、点击C按钮进入该行
相关文章
相关标签/搜索