YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
A4_4410848_Jumanah
This repository contains two simple feedforward neural network models implemented in PyTorch:
- **DummyNet** β a simple feedforward model built without using any activation functions .
- **VanillaNet** β a simple feedforward model built using activation functions.
Both models are trained on the FashionMNIST dataset and their pretrained weights are included.
## Files in this repository
- dummy.py β defines the SimpleFeedForwardNet class for DummyNet
- vanilla.py β defines the SimpleFeedForwardNet class for VanillaNet
- dummy-weights.bin β pretrained weights for DummyNet
- vanilla-weights.bin β pretrained weights for VanillaNet
- hubconf.py β allows loading models with torch.hub.load()
- README.md β this file