.gitignore

@@ -1,4 +1,4 @@
 data
-.venv
+venv
 .vscode
 

quickStart/FashionLabeler.py

@@ -56,7 +56,7 @@ class NeuralNetwork(nn.Module):
         return logits
 
 model = NeuralNetwork()
-model.load_state_dict(torch.load("mdoels/FashionLabeling_model.pth"))
+model.load_state_dict(torch.load("FashionLabeling_model.pth"))
 
 classes = [
     "T-shirt/top",

quickStart/TrainFashionLabeler.py

@@ -101,5 +101,5 @@ for t in range(epochs):
 print("Done!")
 
 
-torch.save(model.state_dict(), "models/model.pth")
+torch.save(model.state_dict(), "model.pth")
 print("Saved PyTorch Model State to model.pth")

requirements.txt

@@ -4,5 +4,4 @@ torch
 torchvision
 --index-url https://download.pytorch.org/whl/cu117
 torchaudio
-numpy
-pandas
+numpy