飞桨封装的resnet50模型,如何直接调用内置层
收藏
如图所示,显示resnet网络结构,如果我要调用Conv2D-372这一层,直接赋予给我自己定义层layer1,从而搭建自己的神经网络,代码应该怎么写呢?
0
收藏
请登录后评论
直接改源码吧:
https://github.com/PaddlePaddle/Paddle/blob/develop/python/paddle/vision/models/resnet.py
用Pycharm这样能够代码提示的IDE,看一下成员变量就可以了,很简单