随笔分类 - 代码 / 算法代码
摘要:容器中pytorch的cpu速度很慢,原因找到了 # import numpy as np # import time # import torch # import os # os.environ['OMP_NUM_THREADS'] = '8' # 增加线程数 # torch.set_num_t
阅读全文
摘要:用pytorch 2.2 快了不少 参考 https://pytorch.org/blog/compiling-numpy-code/ 在mac M2 机器上, 快了50%, 但没有好几倍。可能和依赖libomp有关 brew install libomp python test_np.py tes
阅读全文
摘要:```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
阅读全文

浙公网安备 33010602011771号