Nigel/PytorchTutorials
1
0
Fork 0
Code Releases Activity

Moving FashionLabelingModel into a models folder

Browse Source
This commit is contained in:
Nigel Barink 2023-03-30 19:43:13 +02:00
parent 23a0591630
commit 69089d2af9
3 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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")