The setting problem of TEED training parameters

[KinberAlan]

Hello, sir.
When I use your TEED model for training, I get the.pth file. However, the quality of the prediction results obtained by using the weight file I trained myself is very poor, while using the weight file in your source code can obtain good prediction results. I don 't know if there is a problem with the parameter setting. Moreover, I used the BIPED dataset to train 8 epochs in less than 5 minutes, while the training time mentioned in your article is 30 minutes. I suspect that something must have gone wrong. Can you tell me your training parameter settings ?
The first image is the prediction result obtained by using the weight file obtained by my own training, and the second is the prediction result obtained by using the weight file in your source code.
Hope to receive your reply, thank you very much !

[xavysp]

Hi, thanks for using TEED. looking at the issue 7, I think you fund the problem didn't you?

P. D. What an impressive result.

[KinberAlan]

Hi, thanks for using TEED. looking at the issue 7, I think you fund the problem didn't you?

P. D. What an impressive result.

Yes,!Thank you very much for your attention and answers!
Wish you a good day. I will expand my data set to retrain the model.