Moving FashionLabelingModel into a models folder

Dataloaders and datasets
2023-03-30 19:43:13 +02:00 · 2023-03-30 19:41:46 +02:00
6 changed files with 272 additions and 4 deletions
--- a/.gitignore
+++ b/.gitignore
@ -1,4 +1,4 @@
 data
-venv
+.venv
 .vscode

--- a/data.ipynb
+++ b/data.ipynb
--- a/models/FashionLabeling_model.pth
+++ b/models/FashionLabeling_model.pth
--- a/quickStart/FashionLabeler.py
+++ b/quickStart/FashionLabeler.py
@ -56,7 +56,7 @@ class NeuralNetwork(nn.Module):
        return logits

 model = NeuralNetwork()
-model.load_state_dict(torch.load("FashionLabeling_model.pth"))
+model.load_state_dict(torch.load("mdoels/FashionLabeling_model.pth"))

 classes = [
    "T-shirt/top",
--- a/quickStart/TrainFashionLabeler.py
+++ b/quickStart/TrainFashionLabeler.py
@ -101,5 +101,5 @@ for t in range(epochs):
 print("Done!")


-torch.save(model.state_dict(), "model.pth")
+torch.save(model.state_dict(), "models/model.pth")
 print("Saved PyTorch Model State to model.pth")
--- a/requirements.txt
+++ b/requirements.txt
@ -4,4 +4,5 @@ torch
 torchvision
 --index-url https://download.pytorch.org/whl/cu117
 torchaudio
-numpy
+numpy
+pandas
Author	SHA1	Message	Date
Nigel Barink	69089d2af9	Moving FashionLabelingModel into a models folder	2023-03-30 19:43:13 +02:00
Nigel Barink	23a0591630	Dataloaders and datasets	2023-03-30 19:41:46 +02:00