search

thuml / Transfer-Learning-Library

Transfer Learning Library for Domain Adaptation, Task Adaptation, and Domain Generalization
http://transfer.thuml.ai
MIT License
3.39k stars 553 forks source link

A bug in rsd.py #180

Closed aragakiyui611 closed 1 year ago

aragakiyui611 commented 1 year ago

https://github.com/thuml/Transfer-Learning-Library/blob/7f0bc105f1d8adedf6e8d281a29ced8814d96065/examples/domain_adaptation/image_regression/rsd.py#L87

https://github.com/thuml/Transfer-Learning-Library/blob/7f0bc105f1d8adedf6e8d281a29ced8814d96065/examples/domain_adaptation/image_regression/rsd.py#L91

This two line, the output dim doesn't match. Should change to regressor = Regressor(backbone=backbone, num_factors=num_factors, bottleneck=bottleneck, bottleneck_dim=256).to(device) or something else.

thucbx99 commented 1 year ago

Thanks for pointing this out and we have fix it