This project provides a simple implementation of a feed-forward neural network (also known as a multi-layer perceptron) using PyTorch. The neural network architecture consists of an input layer, a hidden layer with ReLU activation, and an output layer. The purpose of this project is to serve as a basic template for building neural networks in PyTorch.
Python 3.x PyTorch (>= 1.9.0) Installation
Clone this repository to your local machine:
git clone git@github.com:const-antin/ms-git.git
Navigate to the project directory:
cd ms-git
Install the required packages using pip:
pip install -r requirements.txt
Open the simple_nn.py file in your preferred editor. Modify the input_size, hidden_size, and output_size variables in the SimpleNN class constructor according to your requirements. Run the script:
python nn.py
This project is licensed under the MIT License - see the LICENSE file for details.