pgnet.pytorch

pgnet.pytorch copied to clipboard

训练是 ctcloss爆炸

1

ctc loss tensor(inf, device='cuda:0') 0/600/0/2000 | loss:inf | score_loss:0.82 | border_loss:0.56 | direction_loss:0.42 | ctc_loss:inf | | Lr:0.00100000 /home/wzp/project/torch/lib/python3.6/site-packages/torch/nn/functional.py:1569: UserWarning: nn.functional.sigmoid is deprecated. Use torch.sigmoid instead. warnings.warn("nn.functional.sigmoid is deprecated. Use...

apple2333cream

用百度开源的印章数据集，训练到50多个epoch,loss还是没降下去

1

用百度开源的印章数据集（8千张训练集），训练到50多个epoch,loss还是没降下去, 训练日志如下： 53/600/300/333 | loss:103.55 | score_loss:0.03 | border_loss:0.01 | direction_loss:0.01 | ctc_loss:20.70 | | Lr:0.00100000 53/600/305/333 | loss:96.91 | score_loss:0.03 | border_loss:0.01 | direction_loss:0.00 | ctc_loss:19.37 | | Lr:0.00100000...

apple2333cream

pre_trained model

2

thanks for your works ,could you provide a pre_trained model?

Tracy-git

could you provide a pre_trained model? pgnet_model.pth

1

could you provide a pre_trained model? pgnet_model.pth _Originally posted by @Devin521314 in https://github.com/BADBADBADBOY/pgnet.pytorch/issues/1#issuecomment-1241417942_

Devin521314