PyTorch模块的FLOPS计数
用法
通过更改main.py第13行来加载模型
运行python main.py
参考
2022-06-27 08:35:17
2KB
1