zhongyu zhang 57aec1f730

[Feature] Add DAFL Distillation (#235) ...

* 1.Add DAFL, including config, DAFLLoss and readme. 2.Add DataFreeDistillationtillation. 3.Add Generator, including base_generator and dafl_generator. 4.Add get_module_device and set_requires_grad functions in utils.

* 1.Amend the file that report error in mypy test under py37, including gather_tensors, datafree_distillation, base_generator. 2.Revise other linting error.

* 1.Revise some docstrings.

* 1.Add UT for datafreedistillation. 2.Add all typing.hints.

* 1.Add UT for generators and gather_tensors.

* 1.Add assert of batch_size in base_generator

* 1.Isort

Co-authored-by: zhangzhongyu.vendor < zhangzhongyu.vendor@sensetime.com>

2022-08-23 10:47:34 +08:00

..

test_distillation_losses.py

[Feature] Add DAFL Distillation (#235)

2022-08-23 10:47:34 +08:00

test_general_losses.py

[Feature] Add Activation Boundaries Loss (#214)

2022-08-05 14:14:47 +08:00