街景字符識別Task1學習筆記

1.本人使用的筆記本顯卡是GTX1660Ti-MQ 6G,使用GPU跑baseline需要將步驟5的use_cuda改成True。 2.在運行程序時候遇到如下報錯: 原因是GPU裏存的tensor無法轉爲numpy數組,需要先轉爲cpu類型的tensor,解決方案如下: 在data後加.cpu()即可。 3.程序跑通後將epoch設置爲10,其他超參不變的情況下,比較batchsize分別爲30,
相關文章
相關標籤/搜索