clarify restrictions

2020-01-04 22:16:27 +11:00
parent db44e56fb6
commit e6ca3628ce
6 changed files with 21 additions and 18 deletions
--- a/exps/algos/SETN.py
+++ b/exps/algos/SETN.py
@@ -135,6 +135,7 @@ def main(xargs):
  logger = prepare_logger(args)

  train_data, valid_data, xshape, class_num = get_datasets(xargs.dataset, xargs.data_path, -1)
+  assert xargs.dataset == 'cifar10', 'currently only support CIFAR-10'
  if xargs.dataset == 'cifar10' or xargs.dataset == 'cifar100':
    split_Fpath = 'configs/nas-benchmark/cifar-split.txt'
    cifar_split = load_config(split_Fpath, None, None)