Nigel/PytorchTutorials
1
0
Fork 0
Code Releases Activity

Compare commits

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

No commits in common. "69089d2af9abeae3909640763a9d3b9b8013b25f" and "f9fbfae1c11918d81d8dfed6c3bd67e9194e8758" have entirely different histories.
69089d2af9 ... f9fbfae1c1

6 changed files with 4 additions and 272 deletions
Show Stats Expand all files Collapse all files

2
.gitignore vendored
View File

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

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

267
data.ipynb
View File

File diff suppressed because one or more lines are too long

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("mdoels/FashionLabeling_model.pth")) model.load_state_dict(torch.load("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(), "models/model.pth") torch.save(model.state_dict(), "model.pth")
print("Saved PyTorch Model State to model.pth") print("Saved PyTorch Model State to model.pth")

3
requirements.txt
View File

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