Compare commits
2 Commits
f9fbfae1c1
...
69089d2af9
Author | SHA1 | Date |
---|---|---|
Nigel Barink | 69089d2af9 | |
Nigel Barink | 23a0591630 |
|
@ -1,4 +1,4 @@
|
||||||
data
|
data
|
||||||
venv
|
.venv
|
||||||
.vscode
|
.vscode
|
||||||
|
|
||||||
|
|
File diff suppressed because one or more lines are too long
|
@ -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",
|
||||||
|
|
|
@ -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")
|
||||||
|
|
|
@ -4,4 +4,5 @@ torch
|
||||||
torchvision
|
torchvision
|
||||||
--index-url https://download.pytorch.org/whl/cu117
|
--index-url https://download.pytorch.org/whl/cu117
|
||||||
torchaudio
|
torchaudio
|
||||||
numpy
|
numpy
|
||||||
|
pandas
|
Loading…
Reference in New Issue