AI417 - A4
Student: Layan
ID: 4312130
This repository contains two fully connected neural networks trained on FashionMNIST:
- DummyNet
- 4 Linear layers
- No activation functions
- Optimizer: SGD
- Loss: CrossEntropyLoss
- Weights file: dummy_weights.bin
- VanillaNet
- 4 Linear layers
- ReLU activation between hidden layers
- Optimizer: SGD
- Loss: CrossEntropyLoss
- Weights file: vanilla_weights.bin
Models can be loaded using Torch Hub via hubconf.py.
Output dimension: 10 classes.
