thuiar/ch-sims-v2

can't find dataset['train_mix']

wx3771 opened this issue · 3 comments

wx3771 commented

image

Could you please provide this ？ Thank you！
Columbine21 commented

ch-sims-v2/data/load_data.py

Line 40 in 6eb09e5

if self.mode == 'train_mix':

train_mix is an augmentation from the original train set. Using the following operations.

wx3771 commented

q1: when args.supvised_nums default value is 2722, these code will never execute
image

q2: no mask
image

lyh512796310 commented
image

Could you please provide this ？ Thank you！

For the convenience of downloading the newly provided dataset, we divided the supervised data and unsupervised data into two ".pkl" files, and "train_mix” will put the supervised data and unsupervised data together. You can concatenate supervised and unsupervised data together as "train_mix" data. In addition, "mask" is set to 1 for supervised data and 0 for unsupervised data. For example, the data corresponding to MASK: [1,1,0,0] is Data: [supervised data, supervised data, unsupervised data, unsupervised data].