Search on CIFAR10

[youweiliang]

Hi,

Thanks for open-sourcing this work.
I was wondering how to search for a DenseNAS model on CIFAR10 based on ResNet. I have resolved the data loading issue. But for the specification of the search space, I was struggling to make the configuration correct. Because the input size of CIFAR10 is 32*32, I think I need to modify the net_scale, init_dim, and last_dim in imagenet_search_cfg_resnet.yaml. So I remove seven entries in each list in net_scale. But I encountered this error:

Original Traceback (most recent call last):
  File "/usr/local/anaconda3/lib/python3.8/site-packages/torch/nn/parallel/parallel_apply.py", line 61, in _worker
    output = module(*input, **kwargs)
  File "/usr/local/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/home/xx/code/DenseNAS/models/dropped_model.py", line 155, in forward
    betas = [branch_weights[block_id][beta_id]
  File "/home/xx/code/DenseNAS/models/dropped_model.py", line 155, in <listcomp>
    betas = [branch_weights[block_id][beta_id]
IndexError: list index out of range

I greatly appreciate it if you could tell me how to resolve this issue and perform the search on CIFAR10.