ENAS-pytorch

Architecture Discovery Tool

Automates the discovery of neural network architectures by sharing parameters between subgraphs in a large computational graph

PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"

GitHub

3k stars
108 watching
494 forks
Language: Python
last commit: over 1 year ago
google-brainneural-architecture-searchpytorch

Related projects:

Repository Description Stars
pytorch/pytorch A Python library providing tensors and dynamic neural networks with strong GPU acceleration 84,371
yunjey/pytorch-tutorial A comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage. 30,286
dsksd/deepnlp-models-pytorch Pytorch implementations of various Deep NLP models from Stanford's cs-224n course 2,954
lukemelas/efficientnet-pytorch A PyTorch implementation of EfficientNet convolutional neural networks 7,921
pyg-team/pytorch_geometric A PyTorch-based library for training and applying Graph Neural Networks to structured data 21,382
quark0/darts Automated method to find optimal architecture for machine learning models 3,919
diego999/pygat An implementation of the Graph Attention Network model using PyTorch. 2,914
dyhan0920/pyramidnet-pytorch An implementation of a deep neural network architecture for image classification tasks 273
awentzonline/pytorch-cns Search algorithm for neural network optimization 26
ericsun99/shufflenet-v2-pytorch A PyTorch implementation of a ShuffleNet-v2 neural network architecture for image classification. 430
nvidia/flownet2-pytorch A deep learning framework implementing an optical flow estimation model using PyTorch. 3,147
nixtla/neuralforecast An open-source Python library providing a unified interface to various neural forecasting models, focusing on performance, usability, and robustness. 3,101
meetps/pytorch-semseg Implementations of popular semantic segmentation architectures using PyTorch 3,393
qubvel-org/segmentation_models.pytorch A PyTorch library for building and training neural networks for image segmentation tasks. 9,696
d-li14/regnet.pytorch An implementation of a PyTorch-based neural network architecture for image classification tasks. 68