AI417 β Assignment 4 (Swiss Knife)
This repository contains two PyTorch models submitted for AI417 Assignment 4.
Models
- DummyNet: a simple feed-forward network
- VanillaNet: a deeper feed-forward network with LeakyReLU activations
Usage
Models are compatible with torch.hub.load and FashionMNIST-sized inputs (784).
Files
Dummy.pyβ DummyNet architectureVanilla.pyβ VanillaNet architecturehubconf.pyβ HuggingFace hub configurationdummy_weights.binβ DummyNet weightsvanilla_weights.binβ VanillaNet weights
Inference Providers NEW
This model isn't deployed by any Inference Provider. π Ask for provider support