
FLOPS-pt:PyTorch模块的FLOPS计算器
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
FLOPS-pt是一款针对PyTorch开发的计算模型浮点运算次数(FLOPs)工具,帮助开发者精准评估深度学习模型的计算量和性能瓶颈。
使用PyTorch模块进行FLOPS计数的方法是通过更改main.py文件的第13行来加载模型,并运行python main.py命令。
全部评论 (0)
还没有任何评论哟~


简介:
FLOPS-pt是一款针对PyTorch开发的计算模型浮点运算次数(FLOPs)工具,帮助开发者精准评估深度学习模型的计算量和性能瓶颈。
使用PyTorch模块进行FLOPS计数的方法是通过更改main.py文件的第13行来加载模型,并运行python main.py命令。


