Assignment 1 Submission
Used Pytorch pip install torch torchvision torchviz matplotlib
Computational Graph
Input Layer (784)
|
Hidden Layer 1 (128) ---- ReLU
|
Hidden Layer 2 (64) ---- ReLU
|
Output Layer (10) ---- Softmax
Plot results are given in the code