Spaces:
Build error
Build error
fix unused config entries
Browse files- data/build.py +1 -2
data/build.py
CHANGED
@@ -84,8 +84,7 @@ def build_dataset(is_train, config):
|
|
84 |
nb_classes = 1000
|
85 |
elif config.DATA.DATASET == 'inaturelist2021':
|
86 |
root = './datasets/inaturelist2021'
|
87 |
-
dataset = DatasetMeta(root=root,transform=transform,train=is_train,aux_info=config.DATA.ADD_META,dataset=config.DATA.DATASET
|
88 |
-
class_ratio=config.DATA.CLASS_RATIO,per_sample=config.DATA.PER_SAMPLE)
|
89 |
nb_classes = 10000
|
90 |
elif config.DATA.DATASET == 'inaturelist2021_mini':
|
91 |
root = './datasets/inaturelist2021_mini'
|
|
|
84 |
nb_classes = 1000
|
85 |
elif config.DATA.DATASET == 'inaturelist2021':
|
86 |
root = './datasets/inaturelist2021'
|
87 |
+
dataset = DatasetMeta(root=root,transform=transform,train=is_train,aux_info=config.DATA.ADD_META,dataset=config.DATA.DATASET)
|
|
|
88 |
nb_classes = 10000
|
89 |
elif config.DATA.DATASET == 'inaturelist2021_mini':
|
90 |
root = './datasets/inaturelist2021_mini'
|