计算flops和add

最新推荐文章于 2025-11-25 14:31:30 发布

原创最新推荐文章于 2025-11-25 14:31:30 发布 · 306 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #pytorch #人工智能

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))