Concat Dataset giving error - vision - PyTorch Forums
discuss.pytorch.org › t › concat-dataset-givingJun 17, 2020 · Hello Everyone, I have two datasets and want to use them simultaneously while training. The first dataset is a regression dataset containing 5000 while the second dataset is a classification dataset containing 25000 images. The target of the regression dataset is a list containing four numbers while for classification dataset it is a single value depicting the class. I want to train my model ...
【Pytorch】多个数据集联合读取 - 知乎
https://zhuanlan.zhihu.com/p/2227729961、Pytorch的ConcatDataset介绍. class ConcatDataset (Dataset): """ Dataset to concatenate multiple datasets. Purpose: useful to assemble different existing datasets, possibly large-scale datasets as the concatenation operation is done in an on-the-fly manner. Arguments: datasets (sequence): List of datasets to be concatenated ...