pytorch
从数据集中取10条数据进行调试
dataClass.validIdList = dataClass.validIdList[:10]
dataClass.validSampleNum = len(dataClass.validIdList)
dataClass.trainIdList = dataClass.validIdList
dataClass.trainSampleNum = len(dataClass.trainIdList)
pytorch
从数据集中取10条数据进行调试
dataClass.validIdList = dataClass.validIdList[:10]
dataClass.validSampleNum = len(dataClass.validIdList)
dataClass.trainIdList = dataClass.validIdList
dataClass.trainSampleNum = len(dataClass.trainIdList)