Implementation compatible with the recent PyTorch and CUDA version

quark0 / darts

Differentiable architecture search for convolutional and recurrent networks

https://arxiv.org/abs/1806.09055

Apache License 2.0

3.92k stars 843 forks source link

Implementation compatible with the recent PyTorch and CUDA version #169

Open qianjiangcn opened 1 year ago

qianjiangcn commented 1 year ago

Hi all,

The DARTS repo is a great codebase but we found some version issues running it. We made few changes to make it compatible for most recent PyTorch and CUDA for our work. Please feel free to take a look for reference. https://github.com/qianjiangcn/eh-dnas