训练测试集一段时间报错

[ruikai0103]

在训练模型的时候 训练一段时间会出现
UserWarning: Exporting a model to ONNX with a batch_size other than 1, with a variable length with LST
M can cause an error when running the ONNX model with a different batch size. Make sure to save the model with a batch size of 1, or define the initial states (h0/c0) as inputs of the mo
del.

按照教程安装的 包 使用的数据集是 测试二 的那个数据集。

配置文件如下

Model:
CharSet: [' ', S, '4', F, X, '9', E, Q, V, U, '1', J, R, '5', '7', Z, H, G, P,
A, '2', '6', '8', Y, B, I, L, W, K, T, D, C, '3']
ImageChannel: 1
ImageHeight: 32
ImageWidth: -1
Word: false
System:
Allow_Ext: [jpg, jpeg, png, bmp]
GPU: true
GPU_ID: 0
Path: images
Project: my_test
Val: 0.03
Train:
BATCH_SIZE: 16
CNN: {NAME: ddddocr}
DROPOUT: 0.3
LR: 0.01
OPTIMIZER: SGD
SAVE_CHECKPOINTS_STEP: 2000
TARGET: {Accuracy: 0.97, Cost: 0.05, Epoch: 20}
TEST_BATCH_SIZE: 16
TEST_STEP: 1000