- 博客(1)
- 资源 (1)
- 收藏
- 关注
原创 Pytorch测试模型的GFLOPs和Param大小
1. 定义文件flops_counter.pyimport sysimport torchimport torch.nn as nnimport numpy as npdef get_model_complexity_info(model, input_res, print_per_layer_stat=True, ...
2020-02-06 16:50:34 4637 1
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人