pytorch-adapt
pytorch-adapt copied to clipboard
Add domain parameter to CLossHook
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