search

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

Fix freeze unfreez for batch norms #151

Closed PososikTeam closed 2 years ago

PososikTeam commented 2 years ago

A lot of code will be removed if Lightning team merged my PR https://github.com/Lightning-AI/lightning/pull/15063