- 博客(10)
- 收藏
- 关注
原创 几种解决跑深度学习模型时CUDA OUT OF MEMORY:GPU内存报错问题的方法
①使用共享GPU内存;②及时清空显存;③使用检查点存储;④使用混合精度
2024-06-13 23:10:33 2607
原创 yolov5 segment对输出结果进行去框、调节mask透明度以及调节自己想要的掩膜颜色
打开”utils/plots.py”,找到以下代码,并修改alpha,alpha取值范围为0-1,越靠近1越不透明,越靠近0越透明。(只需要将self.black = (0, 0, 0),中的rgb参数(0,0,0)进行修改即可)打开“utils/plots.py",找到以下代码,修改hexs里面的代号就行;二、调节mask透明度。
2024-06-03 14:55:01 296
原创 一种解决pytorch和cuda版本不兼容问题的方法(cuda driver版本:12.0.134)
解决pytorch和cuda版本不兼容问题
2024-05-23 13:21:01 2560 2
原创 一个手撕层次聚类的python代码及其解读(用于理解)
源代码来自 gzh:Johngo学长,本人将修改了一些代码,并解读了一些小白较为难懂的代码,用于学习的过程中对层次聚类的理解。其中,C代表着簇,x代表着簇中的点。
2023-12-03 19:59:35 353
原创 一种解决高版本python无法加载sklearn内置的波士顿房价数据的方法
本人用的是python3.11,在机器学习项目实战时,无法通过传统的方式加载sklearn内置的波士顿房价数据。该错误说明高版本不再适用传统形式加载这个数据集。
2023-11-24 10:51:11 3883 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人