随笔分类 -  代码 / 算法代码

摘要:容器中pytorch的cpu速度很慢,原因找到了 # import numpy as np # import time # import torch # import os # os.environ['OMP_NUM_THREADS'] = '8' # 增加线程数 # torch.set_num_t 阅读全文
posted @ 2025-11-05 16:57 bregman 阅读(34) 评论(0) 推荐(0)
摘要:用pytorch 2.2 快了不少 参考 https://pytorch.org/blog/compiling-numpy-code/ 在mac M2 机器上, 快了50%, 但没有好几倍。可能和依赖libomp有关 brew install libomp python test_np.py tes 阅读全文
posted @ 2023-10-23 19:23 bregman 阅读(172) 评论(0) 推荐(0)
摘要:```python import numpy as np def focalloss(p, y, gamma=5): # https://blog.csdn.net/JNingWei/article/details/80038594 # https://www.cnblogs.com/king-lp 阅读全文
posted @ 2020-08-28 16:14 bregman 阅读(283) 评论(0) 推荐(0)