一个继承nn.module的model它包含一个叫做children()的函数,这个函数可以用来提取出model每一层的网络结构,在此基础上进行修改即可,修改方法如下(去除后两层): resnet_layer = nn.Sequential(*list(model.children())[:-2]) 那么,接下来就可以构建我们的
一个继承nn.module的model它包含一个叫做children()的函数,这个函数可以用来提取出model每一层的网络结构,在此基础上进行修改即可,修改方法如下(去除后两层):
那么,接下来就可以构建我们的网络了:
最后,构建一个对象,并加载resnet预训练的参数就可以啦!
|
2019-06-18
2019-07-04
2021-05-23
2021-05-27
2021-05-27