Attention_ocr.pytorch
Attention_ocr.pytorch copied to clipboard
解码器权重加载写成加载编码器了
if opt.decoder:
print('loading pretrained encoder model from %s' % opt.decoder)
encoder.load_state_dict(torch.load(opt.encoder))
上面这段代码应该是加载decoder, 但其实加载成了encoder,会导致后面测试的时候全是错的