YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
FashionMNIST Feedforward Networks
This repository contains two fully connected neural network architectures trained on the FashionMNIST dataset using PyTorch.
Models
. Dummy Model 1
- Linear layers 4
- No activation functions
- Trained using SGD (lr=0.01)
- CrossEntropyLoss
. Vanilla Model 2
- Linear layers 4
- ReLU activations
- LogSoftmax output
- Trained using SGD (lr=0.1)
- CrossEntropyLoss
Dataset
Both models were trained on the FashionMNIST dataset.
Input size: 784 (28x28 flattened)
Output size: 10 classes
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support