torchvision.models 图片任务加载多个模型,比如resnet,vgg,googlenet

看代码

import torch.nn as nn

import torchvision.models as models
import argparse

parser = argparse.ArgumentParser(description='PyTorch ImageNet Training')

if __name__ == '__main__':

    model_names = sorted(name for name in models.__dict__
                         if name.islower() and not name.startswith("__")
                         and callable(models.__dict__[name]))
    # print(model_names)
    parser.add_argument('-a', '--arch', metavar='ARCH', default='resnet50',
                        choices=model_names,
                        help='model architecture: ' +
                             ' | '.join(model_names) +
                             ' (default: resnet50)')
    args = parser.parse_args()

    model = models.__dict__[args.arch]()

    
    for name, param in model.named_parameters():
   
        print(name,param.size())

resnet50 的参数和形状输出:

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值