Thanks a lot for releasing the code!
When using TSA for domain adaptation in much larger dataset(like Imagenet-1k(source domain)，other dataset(target domain)), I found that the GPU memory is not enougn to deal with the Memory Module(memory_target_features etc. in your code):RuntimeError: CUDA out of memory. It troubles me.
Do you have any good solution for this problem?
Regards。
Thanks a lot for releasing the code!
When using TSA for domain adaptation in much larger dataset(like Imagenet-1k(source domain)，other dataset(target domain)), I found that the GPU memory is not enougn to deal with the Memory Module(memory_target_features etc. in your code):RuntimeError: CUDA out of memory. It troubles me.
Do you have any good solution for this problem?
Regards。