- 博客(5)
- 收藏
- 关注
原创 利用thop计算模型的参数量params和计算量flops
有多个输入时,只需要将input(output,)更改为input(output1,output2,)即可。利用thop计算模型的参数量params和计算量flops。
2023-12-11 17:56:34 203
原创 解决: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 1190 1
原创 Python加速循环的若干个办法
Python加速循环的若干个办法: (1)使用Python封装的函数,而不是自编函数; (2)使用Numpy函数,因为Numpy底层代码语言是C++; 上面两个效果仍不好,尝试下面方法中的一种或多种: (3)将需要处理的图像分为若干个部分,使用multiprocessing多进程并行化,充分利用CPU; (4)采用numba(@njit)加速循环部分(但是与multiprocessing似乎不兼容); (5)采用Cython,修改少量代码,...
2022-05-31 15:07:51 2982
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人