Both models are compatible with torch.hub.load().
-
dummy.py
Contains the architecture definition of the DummyNet model.
-
vanilla.py
Contains the architecture definition of the VanillaNet model.
-
dummy-weights.bin
Trained weights for DummyNet (FashionMNIST).
-
vanilla-weight.bin
Trained weights for VanillaNet (FashionMNIST).
-
hubconf.py
Entry point file required for torch.hub.load().
It defines how the models are constructed and how weights are loaded.