1
0
Fork 0

Moving FashionLabelingModel into a models folder

main
Nigel Barink 2023-03-30 19:43:13 +02:00
parent 23a0591630
commit 69089d2af9
3 changed files with 2 additions and 2 deletions

View File

@ -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",

View File

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