时间戳

首页 > 解决方案 > Does a 'ConcatDataset' in pytorch iterates each Datasets ununiformly?

pytorch - Does a 'ConcatDataset' in pytorch iterates each Datasets ununiformly?

问题描述

I have two custom Dataset, and use ConcatDataset for concat these Dataset during training. and I use this code for concat dataset = ConcatDataset([CustomDataset1,CustomDataset2])

and this ConcatDataset pick each dataset one by one uniformly. but I want to pick each dataset un-uniformly

ex. CustomDataset1, CustomDataset1, CustomDataset2, CustomDataset1, CustomDataset1, CustomDataset2 ...

How can I that???

标签： pytorch

解决方案

推荐阅读