crnn.pytorch icon indicating copy to clipboard operation
crnn.pytorch copied to clipboard

how to modify train.py ?

Open OKhyc opened this issue 6 years ago • 1 comments

i only have CPU environment. how to modify the code of train.py that can train chinese Character? currently i have 5000 images .

OKhyc avatar Jan 05 '19 06:01 OKhyc

GPU 并不是必须的。 可以搞个 flag_cuda=torch.cuda.is_available() 然后基于这个 flag_cuda决定是否要对 model, tensor等 进行 cuda() 调用

leonzhu211 avatar Jan 24 '19 06:01 leonzhu211