API to compose PyTorch neural networks on the fly.
Project description
Torch-Nets
Compose PyTorch neural networks with ease.
Installation
From PYPI (current version: v0.0.1
)
pip install torch-nets
Alternatively, install the development version from GitHub:
git clone https://github.com/mvinyard/torch-nets.git;
cd torch-nets; pip install -e .
Example API use-case
from torch_nets import TorchNet
Create a feed-forward neural network
net = TorchNet(
in_features=50,
out_features=50,
hidden=[400, 400],
activation="LeakyReLU",
dropout=0.2,
n_augment=0,
bias=True,
output_bias=True,
)
net
Sequential(
(hidden_1): Sequential(
(linear): Linear(in_features=50, out_features=400, bias=True)
(dropout): Dropout(p=0.2, inplace=False)
(activation): LeakyReLU(negative_slope=0.01)
)
(hidden_2): Sequential(
(linear): Linear(in_features=400, out_features=400, bias=True)
(dropout): Dropout(p=0.2, inplace=False)
(activation): LeakyReLU(negative_slope=0.01)
)
(output): Sequential(
(linear): Linear(in_features=400, out_features=50, bias=True)
)
)
The only required arguments are in_features
and out_features
. The network can be made as simple or complex as you want through optional parameters.
Potential future plans
- Composition of
torch.optim
funcs.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
torch-nets-0.0.2.tar.gz
(19.3 kB
view hashes)
Built Distribution
torch_nets-0.0.2-py3-none-any.whl
(22.2 kB
view hashes)
Close
Hashes for torch_nets-0.0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1843273e263b2fb59dc33058cc5b135ff9e93fd200f37bc6bfaf32f387851fe8 |
|
MD5 | db8784699df12dfc8bd56a8a8a00153d |
|
BLAKE2b-256 | 26b28774b0533f12189b69a9b11cf5b39958254bfd8d58c9f2eb23ab4f10f414 |