1
0

Moving FashionLabelingModel into a models folder

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

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

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