Nigel/PytorchTutorials
1
0
Fork 0
Code Releases Activity

Compare commits

base: Nigel:f9fbfae1c11918d81d8dfed6c3bd67e9194e8758
Branches Tags
Nigel:main
...
compare: Nigel:69089d2af9abeae3909640763a9d3b9b8013b25f
Branches Tags
Nigel:main

2 Commits
f9fbfae1c1 ... 69089d2af9

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
6 changed files with 272 additions and 4 deletions
Expand all files Collapse all files

2
.gitignore vendored
View File

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

267
data.ipynb Normal file
View File

File diff suppressed because one or more lines are too long

0
FashionLabeling_model.pth → models/FashionLabeling_model.pth
View File

2
quickStart/FashionLabeler.py
View File

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

2
quickStart/TrainFashionLabeler.py
View File

@ -101,5 +101,5 @@ for t in range(epochs):
print("Done!") 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") print("Saved PyTorch Model State to model.pth")

1
requirements.txt
View File

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