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.
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support