search

KevinMusgrave / pytorch-adapt

Domain adaptation made easy. Fully featured, modular, and customizable.
https://kevinmusgrave.github.io/pytorch-adapt/
MIT License
360 stars 15 forks source link

Add domain parameter to CLossHook #68

Closed KevinMusgrave closed 2 years ago

KevinMusgrave commented 2 years ago

Right now it's hardcoded to use src_logits. Adding a domain parameter (set to either src or target) would allow CLossHook to be used for supervised domain adaptation as well.

https://github.com/KevinMusgrave/pytorch-adapt/blob/3b2713c4860b325c79481f11307a193bb381d53f/src/pytorch_adapt/hooks/classification.py#L75-L88

KevinMusgrave commented 2 years ago

Now in v0.0.82