关于自定义网络问题

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

Closed JiaQi1996 closed 1 year ago

JiaQi1996 commented 1 year ago

作者您好，非常感谢您能够构建TLlib。在使用TLlib的过程中，我想要自定义一个多分枝网络（类似于BBN或使用多个分类器），请问TLlib能否支持自定义网络。

thucbx99 commented 1 year ago

可以的，你只需要重写Classifier类即可，我们目前实现的方法中也有一些需要多分枝网络，比如MDD中实现的有main head和adversarial head的module，你可以参考一下

JiaQi1996 commented 1 year ago

感谢