search

robot-learning-freiburg / MM-DistillNet

PyTorch code for training MM-DistillNet for multimodal knowledge distillation. http://rl.uni-freiburg.de/research/multimodal-distill
GNU General Public License v3.0
59 stars 14 forks source link

The thermal teacher #15

Open LE0J-Song opened 2 years ago

LE0J-Song commented 2 years ago

Hi,

Thanks for sharing this great work!

I just meet some problems about the thermal teacher. When I load the thermal teacher model, I found that most models were not updated to the initialized efficientDet. The evaluate.log output is :

using path=trained_models/yet-another-efficientdet-d2-thermal.pth ModelDict Update:174/1076

Because I found that the thermal teacher's batch_labels is always empty.

So can you teach me how to load the thermal teacher model correctly?

Thank you very much!