search

carpedm20 / ENAS-pytorch

PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"
Apache License 2.0
2.69k stars 492 forks source link

changes to run on PyTorch version >= 0.4.0 #26

Closed philtomson closed 5 years ago

philtomson commented 5 years ago

The current code only runs on versions of PyTorch before 0.4.0, this pull request adds changes to allow running on PyTorch versions >= 0.4.0.

see: https://github.com/carpedm20/ENAS-pytorch/issues/25

carpedm20 commented 5 years ago

Thanks for the PR!