摘要:
import torch from flops_counter import get_model_complexity_info input_shape = (3, 320, 568) split_line = '=' * 30 flops, params = get_model_complexit 阅读全文
posted @ 2022-04-17 15:42
咖啡陪你
阅读(420)
评论(0)
推荐(0)
摘要:
在pytorch环境下,有两个计算FLOPs和参数量的包thop和ptflops,结果基本是一致的。 thop 参考https://github.com/Lyken17/pytorch-OpCounter 安装方法:pip install thop 使用方法: from torchvision.mo 阅读全文
posted @ 2022-04-17 15:40
咖啡陪你
阅读(2981)
评论(0)
推荐(0)

浙公网安备 33010602011771号