pytorch-adapt icon indicating copy to clipboard operation
pytorch-adapt copied to clipboard

Add domain parameter to CLossHook

Open KevinMusgrave opened this issue 2 years ago • 0 comments

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 avatar May 02 '22 19:05 KevinMusgrave