1.本人使用的笔记本显卡是GTX1660Ti-MQ 6G,使用GPU跑baseline需要将步骤5的use_cuda改成True。
2.在运行程序时候遇到如下报错:街景字符识别Task1学习笔记
原因是GPU里存的tensor无法转为numpy数组,需要先转为cpu类型的tensor,解决方案如下:街景字符识别Task1学习笔记
在data后加.cpu()即可。
3.程序跑通后将epoch设置为10,其他超参不变的情况下,比较batchsize分别为30,40,64时模型迭代10后的效果发现batchsize=40时效果最好,可以在batchsize为40与64之间多取几个值试试。

batchsize=30
街景字符识别Task1学习笔记
batchsize=40
街景字符识别Task1学习笔记
batchsize=64街景字符识别Task1学习笔记

相关文章:

  • 2021-08-30
  • 2021-12-28
  • 2021-05-14
  • 2021-04-12
  • 2021-10-03
  • 2022-01-13
  • 2021-07-31
  • 2022-01-16
猜你喜欢
  • 2021-10-14
  • 2021-07-20
  • 2021-10-16
  • 2021-07-11
  • 2021-10-09
  • 2021-10-19
  • 2021-06-17
相关资源
相似解决方案