Closed smilesun closed 1 month ago
Merge pull request https://github.com/marrlab/DomainLab/pull/824 from marrlab/irm c5646fa · 3 minutes ago HistoryMerge pull request https://github.com/marrlab/DomainLab/pull/824 from marrlab/irm c5646fa · 3 minutes ago History
paper: https://arxiv.org/pdf/1907.02893 paper section to read: 3.2 Implementation details When estimating the objective (IRMv1) using mini-batches for stochastic gradient descent, one can obtain an unbiased estimate of the squared gradient norm as
reference code:
https://github.com/facebookresearch/DomainBed/blob/93ebb6bccdb46d40471ce0713a165aaf7b5e21b6/domainbed/algorithms.py#L312