dxywill / nas_rl
Pytorch Implementation of Neural Architecture Search with Reinforcement Learning (in dev)
☆19Updated 4 years ago
Related projects: ⓘ
- Open source reproduction in PyTorch of "Neural Predictor for Neural Architecture Search".☆38Updated 4 years ago
- [ICLR2021 Outstanding Paper] Rethinking Architecture Selection in Differentiable NAS☆102Updated 2 years ago
- TPAMI 2021: NATS-Bench: Benchmarking NAS Algorithms for Architecture Topology and Size☆171Updated last year
- ☆74Updated 5 months ago
- [ICLR-2020] Dynamic Sparse Training: Find Efficient Sparse Network From Scratch With Trainable Masked Layers.☆31Updated 4 years ago
- [NeurIPS 2020] "Does Unsupervised Architecture Representation Learning Help Neural Architecture Search?" by Shen Yan, Yu Zheng, Wei Ao, X…☆49Updated 3 years ago
- ☆25Updated 3 months ago
- PyTorch port of "Efficient Neural Architecture Search via Parameters Sharing"☆52Updated 6 years ago
- [ICML 2021] "Do We Actually Need Dense Over-Parameterization? In-Time Over-Parameterization in Sparse Training" by Shiwei Liu, Lu Yin, De…☆46Updated 10 months ago
- This repo contains the PyTorch implementation of the SNAS-Series papers☆145Updated 3 years ago
- reimplementation of "Learning Transferable Architectures for Scalable Image Recognition" using mnist dataset, include controller☆42Updated 4 years ago
- Soft Threshold Weight Reparameterization for Learnable Sparsity☆88Updated last year
- ☆35Updated 3 years ago
- Pytorch Implementation of Neural Architecture Optimization☆111Updated 3 years ago
- Pytorch implementation of the paper "SNIP: Single-shot Network Pruning based on Connection Sensitivity" by Lee et al.☆101Updated 5 years ago
- Code for our ICML'2020 paper "Stabilizing Differentiable Architecture Search via Perturbation-based Regularization"☆75Updated 3 years ago
- MiLeNAS: Efficient Neural Architecture Search via Mixed-Level Reformulation. Published in CVPR 2020☆37Updated 3 years ago
- Reproducing RigL (ICML 2020) as a part of ML Reproducibility Challenge 2020☆26Updated 2 years ago
- [ICLR 2021] "Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective" by Wuyang Chen, Xinyu Gong, …☆164Updated 2 years ago
- Zero-Cost Proxies for Lightweight NAS☆141Updated last year
- Implementation of Effective Sparsification of Neural Networks with Global Sparsity Constraint☆28Updated 2 years ago
- Measuring and predicting on-device metrics (latency, power, etc.) of machine learning models☆66Updated last year
- [ICLR 2021] HW-NAS-Bench: Hardware-Aware Neural Architecture Search Benchmark☆101Updated last year
- [ICML2022] Training Your Sparse Neural Network Better with Any Mask. Ajay Jaiswal, Haoyu Ma, Tianlong Chen, ying Ding, and Zhangyang Wang☆23Updated 2 years ago
- ☆150Updated 4 years ago
- Homogeneous Architecture Augmentation for Neural Predictor☆8Updated 3 years ago
- Official PyTorch Implementation for Conflict-Averse Gradient Descent (CAGrad)☆108Updated 10 months ago
- [AAAI '23] PINAT: A Permutation INvariance Augmented Transformer for NAS Predictor☆29Updated last year
- [Neurips 2021] Sparse Training via Boosting Pruning Plasticity with Neuroregeneration☆27Updated last year
- Source Code for ICML 2019 Paper "Shallow-Deep Networks: Understanding and Mitigating Network Overthinking"☆35Updated 8 months ago