search

lixiaotong97 / DSU

[ICLR 2022] Official pytorch implementation of "Uncertainty Modeling for Out-of-Distribution Generalization" in International Conference on Learning Representations (ICLR) 2022.
142 stars 16 forks source link

about multi-domain-generalization #8

Open Xiloy opened 1 year ago

Xiloy commented 1 year ago

Can I use this to train voc dataset

lixiaotong97 commented 1 year ago

Hi, thanks for your interests. I don't try voc before, but I think DSU also works to improve the generlization ability. Hyper-parameter such as p and inserted positions can be tuned, e.g., setting p=0.2, inserted position=0,1,2,3,4, etc to suit the task.

Xiloy commented 1 year ago

Thanks a lot, btw can you give me some advice or general direction on where this module should be applied in the network(yolov5)，eg, in shallow or deep network…