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

解码器权重加载写成加载编码器了

Open ly0303521 opened this issue 5 years ago • 0 comments

if opt.decoder:
    print('loading pretrained encoder model from %s' % opt.decoder)
    encoder.load_state_dict(torch.load(opt.encoder))

上面这段代码应该是加载decoder, 但其实加载成了encoder,会导致后面测试的时候全是错的

ly0303521 avatar May 20 '20 06:05 ly0303521