Moving FashionLabelingModel into a models folder
This commit is contained in:
parent
23a0591630
commit
69089d2af9
@ -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",
|
||||
|
@ -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")
|
||||
|
Loading…
Reference in New Issue
Block a user