I have some questions about training and testing

[alan-img]

   First and foremost, thanks to you opensource the awesome STR algorithm,I have some questions about training and testing.

I cope the checkpoint from the docker image, And I find detecting and recognition separted two model, one is ICDAR_0.7.pb, another is text_recognition_5435.pb. this mode is not apropriate this repo code. Hence I train my model through dataset.py genderating icdar_datasets.npy , with a lot of reading code, I find that the test.py can not work.it appear the error:
2021-06-16 15:01:59.018589: I tensorflow/stream_executor/platform/default/dso_loader.cc:42] Successfully opened dynamic library libcudnn.so.7 2021-06-16 15:01:59.233909: E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR 2021-06-16 15:01:59.261653: E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR Traceback (most recent call last): File "/home/alan/anaconda3/envs/AttentionOCR/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1356, in _do_call return fn(*args) File "/home/alan/anaconda3/envs/AttentionOCR/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1341, in _run_fn options, feed_dict, fetch_list, target_list, run_metadata) File "/home/alan/anaconda3/envs/AttentionOCR/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1429, in _call_tf_sessionrun run_metadata) tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found. (0) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [[{{node InceptionV4/InceptionV4/Conv2d_1a_3x3/Conv2D}}]] (1) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [[{{node InceptionV4/InceptionV4/Conv2d_1a_3x3/Conv2D}}]] [[sequence_preds/_17]]