zhaolimin

第一是增加batch size,增加GPU的内存占用率,尽量用完内存,而不要剩一半,空的内存给另外的程序用,两个任务的效率都会非常低。

第二,在数据加载时候,将num_workers线程数设置稍微大一点,推荐是8,16等,且开启pin_memory=True。不要将整个任务放在主进程里面做,这样消耗CPU,且速度和性能极为低下。

分类:

技术点:

相关文章:

  • 1970-01-01
  • 2022-12-23
  • 2021-11-20
  • 2021-04-28
  • 2022-12-23
  • 2022-01-17
  • 2021-12-07
  • 2021-11-30
猜你喜欢
  • 2021-12-07
  • 2021-08-03
  • 2021-04-28
  • 2021-12-07
  • 2022-01-16
  • 2021-12-07
  • 2021-05-21
相关资源
相似解决方案