老师，这是显存不够的意思么

Question

Answer 1

会写代码的好厨师 2022-08-17 23:12:49

可以直接用我训练好的模型，我应该在代码仓库里面有文档说明。

• 提问者 慕工程0291652 #1

  老师你都训练了哪些数据集，我使用的时候好有个底

Answer 2

会写代码的好厨师 2022-08-17 22:38:57

显存不够，2g显存训练这个模型不够的，直接用cpu训练吧。或者把网络尺寸改小点，精度和收敛情况可能有影响。

• 提问者 慕工程0291652 #1

  用CPU，一天2个epoch！！我修改batch size等于32，16，也不行，一样的错误，你的说网络尺寸是指哪个？

• 提问者 慕工程0291652 #2

  我可以直接用你的训练好的模型吗？？用于开发自己想要的接口

Answer 3

提问者 慕工程0291652 2022-08-17 22:01:34

我自己笔记本显卡是MX250，2G的，

per_process_gpu_memory_fraction

这个参数也改过，也不行一直报上面的错误