diff --git a/FashionLabeling_model.pth b/models/FashionLabeling_model.pth similarity index 100% rename from FashionLabeling_model.pth rename to models/FashionLabeling_model.pth diff --git a/quickStart/FashionLabeler.py b/quickStart/FashionLabeler.py index a68bcff..4ab856b 100644 --- a/quickStart/FashionLabeler.py +++ b/quickStart/FashionLabeler.py @@ -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", diff --git a/quickStart/TrainFashionLabeler.py b/quickStart/TrainFashionLabeler.py index 2f2d135..9c1cbb8 100644 --- a/quickStart/TrainFashionLabeler.py +++ b/quickStart/TrainFashionLabeler.py @@ -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")