计算flops和add

关注

原创最新推荐文章于 2026-04-10 10:00:00 发布 341 阅读 AI 写同款· GEO 优化 ›

整理经典网络并计算其flops等利用tensorboard显示模型示意图使用到的函数add_graph中的参数 add_graph(model, input_to_model=None, verbose=False, **kwargs) 参数 model (torch.nn.Module): 待可视化的网络模型 input_to_model (torch.Tensor or list of torch.Tensor, optional): 待输入神经网络的变量或一组变量 verbose表示详细信息，verbos... 阅读详情

from model import dfsfne
import torch
from ptflops import get_model_complexity_info
from option import args

with torch.cuda.device(0):
  net  = dfsfne.DFSFN(args)
  flops, params = get_model_complexity_info(net, (3, 360, 240), as_strings=True, print_per_layer_stat=True)
  print('{:<30}  {:<8}'.format('Computational complexity: ', flops))
  print('{:<30}  {:<8}'.format('Number of parameters: ', params))