bonlime / pytorch-tools

Tool box for PyTorch
MIT License
186 stars 16 forks source link

fix loading effdet with coco weights #103

Closed bonlime closed 3 years ago

bonlime commented 3 years ago

Closes #101 Проблема была bias в BiFPN head, который по сути не нужен, но есть в весах. мне слишком лень удалять его из весов и заново их заливать, поэтому просто загружаю с strict=False, надеюсь это не вызовет проблем в будущем