marrlab / DomainLab

modular domain generalization: https://pypi.org/project/domainlab/
https://marrlab.github.io/DomainLab/
MIT License
40 stars 2 forks source link

write documentation for implementing new models and trainers #802

Closed smilesun closed 2 months ago

smilesun commented 3 months ago

In DomainLab, a model specifies $\ell(\cdot) + \mu R(\cdot)$ loss via auxilliary neural network, see https://github.com/marrlab/DomainLab/tree/master/domainlab/models where the auxiliary neural network is correponding to $R(\cdot)$, there exist https://github.com/marrlab/DomainLab/blob/master/domainlab/models/model_custom.py, but this is not a concrete example, we need a documentation/tutorial in the form of a markdown file to guide the user to define their own model.