首页 > 其他分享 >11

11

时间:2023-11-20 10:01:49浏览次数:26  
标签:11 224 params print import model flops

# -- coding: utf-8 --
import torch
import torchvision
from thop import profile

# Model
print('==> Building model..')
model = torchvision.models.alexnet(pretrained=False)

dummy_input = torch.randn(1, 3, 224, 224)
flops, params = profile(model, (dummy_input,))
print('flops: ', flops, 'params: ', params)
print('flops: %.2f M, params: %.2f M' % (flops / 1000000.0, params / 1000000.0))

 

 

# -- coding: utf-8 --
import torchvision
from ptflops import get_model_complexity_info

model = torchvision.models.alexnet(pretrained=False)
flops, params = get_model_complexity_info(model, (3, 224, 224), as_strings=True, print_per_layer_stat=True)
print('flops: ', flops, 'params: ', params)

 

标签:11,224,params,print,import,model,flops
From: https://www.cnblogs.com/yyhappy/p/17843287.html

相关文章

  • 11.15 监控目录文件变化
    监视对指定目录的更改,并将有关更改的信息打印到控制台,该功能的实现不仅可以在内核层,在应用层同样可以。程序中使用ReadDirectoryChangesW函数来监视目录中的更改,并使用FILE_NOTIFY_INFORMATION结构来获取有关更改的信息。ReadDirectoryChangesW是Windows操作系统提供的一个函数,......
  • 每周总结11.01
    安装并学习了基础的vuebuild项目构建(webpack)相关代码config 配置目录,包括端口号等。我们初学可以使用默认的。node_modules npm加载的项目依赖模块assets:放置一些图片,如logo等。components:目录里面放了一个组件文件,可以不用。App.vue:项目入口文件,我们也可以直接将组件......
  • 11.20每日总结
    今天早上进行了软件设计模式的实验16实验16:命令模式本次实验属于模仿型实验,通过本次实验学生将掌握以下内容: 1、理解命令模式的动机,掌握该模式的结构;2、能够利用命令模式解决实际问题。 [实验任务一]:多次撤销和重复的命令模式某系统需要提供一个命令集合(注:可以使用链表,栈......
  • win11笔记本换内存后,报错,及解决:0x00007FF8011F6693指令引用了0x0000000000000000内存
    笔记本原装内存为一对镁光8GDDR54800MHz换单条镁光32GDDR55600MHz内存后,重启电脑出现如下报错:0x00007FF8011F6693指令引用了0x0000000000000000内存。该内存不能为read。要终止程序,请单击”确定” 联系内存的卖家客服提供的解决步骤虽然我没看到滚屏,但是重启后问题一样......
  • 11月20每日打卡
    [实验任务一]:JAVA和C++常见数据结构迭代器的使用信1305班共44名同学,每名同学都有姓名,学号和年龄等属性,分别使用JAVA内置迭代器和C++中标准模板库(STL)实现对同学信息的遍历,要求按照学号从小到大和从大到小两种次序输出学生信息。实验要求:1. 搜集并掌握JAVA和C++中常见的数据结构......
  • 每日总结-23.11.18
    publicclassStudent{privateStringid;privateStringname;privateStringage;publicStringgetAge(){returnage;}publicvoidsetAge(Stringage){this.age=age;}publicStudent(){}publicStude......
  • 每日总结-23.11.19
    importjava.util.Stack;publicabstractclassAbstractCommand{publicabstractintexecute(intvalue);publicabstractintundo();publicabstractintredo();}publicclassAddCommandextendsAbstractCommand{privateAdderadder=newA......
  • 每日总结-23.11.20
    publicclassMemory{privatestaticMemoryinstance=newMemory();publicstaticMemorygetInstance(){returninstance;}publicbooleancheck(booleanoperIf){System.out.println("内存开始自检");if(ope......
  • 20231109学习总结
    MySQL重新初始化1.需要将mysql数据目录下的数据全部清空rm-rf/data/mysql/data 2.初始化数据库/usr/local/mysql/bin/mysqld   --initialize--user=mysql--basedir=/usr/local/mysql--datadir=/data/mysql/data/ 3.重新启动数据库 /etc/init.d/mysqldst......
  • 20231120学习总结.
    信1305班共44名同学,每名同学都有姓名,学号和年龄等属性,分别使用JAVA内置迭代器和C++中标准模板库(STL)实现对同学信息的遍历,要求按照学号从小到大和从大到小两种次序输出学生信息。Java:publicinterfaceAggregate{publicvoidadd(Objectobj);publicvoidremove(Ob......