MichelBartels
|
84147edcca
|
Model Distillation (#1758)
* initial commit
* Add latest docstring and tutorial changes
* added comments and fixed bug
* fixed bugs, added benchmark and added documentation
* Add latest docstring and tutorial changes
* fix type: ignore comment
* fix logging in benchmark
* fixed distillation config
* Add latest docstring and tutorial changes
* added type annotations
* fixed distillation loss calculation
* added type annotations
* fixed distillation mse loss
* improved model distillation benchmark config loading
* added temperature for model distillation
* removed uncessary imports, added comments, added named parameter calls
* Add latest docstring and tutorial changes
* added some more comments
* added distillation test
* fixed distillation test
* removed unnecessary import
* fix softmax dimension
* add grid search
* improved model distillation benchmark config
* fixed model distillation hyperparameter search
* added doc strings and type hints for model distillation
* Add latest docstring and tutorial changes
* fixed type hints
* fixed type hints
* fixed type hints
* wrote out params instead of kwargs in DistillationDataSilo initializer
* fixed type hints
* fixed typo
* fixed typo
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
|
2021-11-26 18:49:30 +01:00 |
|