- 博客(5)
- 收藏
- 关注
RSS订阅原创 利用thop计算模型的参数量params和计算量flops
有多个输入时，只需要将input(output,)更改为input(output1,output2,)即可。利用thop计算模型的参数量params和计算量flops。
2023-12-11 17:56:34
159
原创 解决：CUDA out of memory: Tried to allocate 12.11 GiB......
解决：运行推理前：export PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:32。
2023-07-03 13:32:55
1151
1
原创 Python加速循环的若干个办法
Python加速循环的若干个办法： （1）使用Python封装的函数，而不是自编函数； （2）使用Numpy函数，因为Numpy底层代码语言是C++； 上面两个效果仍不好，尝试下面方法中的一种或多种： （3）将需要处理的图像分为若干个部分，使用multiprocessing多进程并行化，充分利用CPU； （4）采用numba(@njit)加速循环部分（但是与multiprocessing似乎不兼容）； （5）采用Cython，修改少量代码，...
2022-05-31 15:07:51
2960
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人