Output files are empty when testing

Hello, thank you so much for providing code for great research paper. I am just wondering after I run python train.py --config config.json. All the values of mEP, mER, mEF, mEA are all 0 for every train Epoch. Is this normal? 2020-07-26 20:45:35.357166: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library libcudart.so.10.1 [2020-07-26 20:45:40,891 - train - INFO] - Model trainable parameters: 68575842 [2020-07-26 20:45:40,892 - train - INFO] - Train datasets: 8 samples Validation datasets: 8 samples Max_epochs: 100 Log_per_step: 10 Validation_per_step: 50 [2020-07-26 20:45:40,892 - train - INFO] - Training start… [2020-07-26 20:46:07,680 - trainer - INFO] - Train Epoch:[1/100] Step:[2/2] Total Loss: 1121.992798 GL_Loss: 8.390446 CRF_Loss: 1113.602295 [2020-07-26 20:46:09,988 - trainer - INFO] - [Epoch Validation] Epoch:[1/100] Total Loss: 1430.095032 GL_Loss: 0.091890 CRF_Loss: 1420.906006 ±--------±------±------±------±------+ | name | mEP | mER | mEF | mEA | +=========+=======+=======+=======+=======+ | date | 0 | 0 | 0 | 0 | ±--------±------±------±------±------+ | name | 0 | 0 | 0 | 0 | ±--------±------±------±------±------+ | overall | 0 | 0 | 0 | 0 | ±--------±------±------±------±------+ Also, I ran

python test.py --checkpoint /content/PICK-pytorch/saved/models/PICK_Default/test2_0726_194721/model_best.pth --boxes_transcripts /content/PICK-pytorch/data/test_data_example/boxes_and_transcripts \
               --images_path /content/PICK-pytorch/data/test_data_example/images/ --output_folder /content/PICK-pytorch/output/test2 \
               --gpu 0 --batch_size 2

Output files are generated, but the files are actually empty. Can you please give me the right direction to test the model? Thank you!

[wenwenyu]

@sophiajwchoi Theoretically, this model is also compatible with other types of documents. All you need to do is to gather as many data and label as possible.

[sophiajwchoi]

Thank you for replying! The issue has been resolved. Just to confirm, this model can be used to train other form documents.