iSEE-Laboratory/DiffUIR

关于训练集问题

Closed this issue · 2 comments

你好!想问下你的训练集中是把不同任务的数据都放进去了吗,比如含雨、雾、阴影这种不同类型的数据,是否是在一次训练中全都放进去呢?

是的,我们每个batch会同时包含所有类型的退化数据,同时会按照不同退化任务数据集的比例分配每个batch中的数量,具体代码在src/model.py Line1208