自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(3)
  • 问答 (1)
  • 收藏
  • 关注

原创 关于训练模型时num_workers和batch_size的设置匹配

我的想法是batch_size调低,gpu处理一次batch就更加轻松,对下一个batch的需求就更加渴望,那么负责dataloader的cpu就会不停歇地处理和加载数据喂给gpu。所以干脆把batch_size拉到很高,让gpu处理一批数据的速度不要太快,那么cpu加载数据也就更加游刃有余。我的cpu是i7,gpu是4090d,之前跑代码一直会蓝屏,按照网上的意见调低batch_size和num_workers。发现batch_size越低,cpu占用率越高,而gpu占用率百分之一都不到。

2024-07-09 09:11:00 90

原创 求助:RuntimeError: CUDA error: unknown error

以前能跑完训练的代码,上周开始跑到一半就会报错,batch_size调小了一半也不行。

2024-07-08 10:23:09 408 1

原创 求CUHK-PEDES数据集

如题。

2023-10-09 15:03:56 329 7

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除