Attention_ocr.pytorch
Attention_ocr.pytorch copied to clipboard
This repository implements the the encoder and decoder model with attention model for OCR
你好,我在运行你的代码时候报错KeyError:' ', 这种是怎么回事呀? if isinstance(text, str): text = [self.dict[item] for item in text]
您好,大佬问您一个问题,为什么attention解码训练的时候,都要重置 decoder_hidden = decoder.initHidden(b).cuda()参数呢,我的理解应该是编码层输出会有一个decoder_hidden 参数啊,大佬可以解答一下吗? @chenjun2hao
hi, thanks your excellent job, I meet the error: RuntimeError: Error(s) in loading state_dict for decoder: size mismatch for decoder.embedding.weight: copying a param of torch.Size([17765, 256]) from checkpoint, where the...
感觉就取了前几张训练,loss下降的很快,然后test都为空
模型准确率的问题
你好,我想问一下,我重新运行了一下你的代码但是21轮epoch之后,识别的准确率还是很低,达不到你所给出的效果。您觉得,这可能与什么原因有关呢?
Remove consecutive "the" occurrence
``` if opt.decoder: print('loading pretrained encoder model from %s' % opt.decoder) encoder.load_state_dict(torch.load(opt.encoder)) ``` 上面这段代码应该是加载decoder, 但其实加载成了encoder,会导致后面测试的时候全是错的
1.固定2行,第一行4个字,第2行7个字,在不分隔的情况下能使用attention训练吗 2.试了ctc不行