This repository contains the tutorial code for Building a Feedforward Neural Network using Pytorh NN Module
- Import libraries
- Generate non-linearly separable data
- Feedforward network using tensors and auto-grad
- Train our feedforward network
- NN.Functional
- NN.Parameter
- NN.Linear and Optim
- NN.Sequential
- Moving the Network to GPU
- Follow this blog for the detailed explanation of code. Building a Feedforward Neural Network using Pytorch NN Module