Fails when loading models the trained on custom dataset

[rose-jinyang]

Hello How are you? Thanks contributing to this project. I trained a model using 4 GPUs and 5 workers. I did NOT change any parameters except dataset path. I got the following issue when loading the trained models in test.py script.

[zxd-cqu]

the same problem。did you fixed it ？