AI417 - A4

Student: Layan ID: 4312130

This repository contains two fully connected neural networks trained on FashionMNIST:

  1. DummyNet
  • 4 Linear layers
  • No activation functions
  • Optimizer: SGD
  • Loss: CrossEntropyLoss
  • Weights file: dummy_weights.bin
  1. 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.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support