Freeze-Unfreeze callback doesn't freeze batch norm

eora-ai / torchok

Production-oriented Computer Vision models training pipeline for common tasks: classification, segmentation, detection and representation🥤

https://torchok.readthedocs.org

Apache License 2.0

54 stars 6 forks source link

Closed PososikTeam closed 2 years ago

PososikTeam commented 2 years ago

Describe the bug Freeze-Unfreeze callback doesn't freeze batch norms