qubvel-org / segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
https://smp.readthedocs.io/
MIT License
9.74k stars 1.68k forks source link

Fix hub_mixin.py pop error #909

Closed ytzfhqs closed 3 months ago

ytzfhqs commented 3 months ago

Fix the error caused by pop operation on non-existing key in dict.

qubvel commented 3 months ago

Thanks!