From 69089d2af9abeae3909640763a9d3b9b8013b25f Mon Sep 17 00:00:00 2001 From: Nigel Barink Date: Thu, 30 Mar 2023 19:43:13 +0200 Subject: [PATCH] Moving FashionLabelingModel into a models folder --- FashionLabeling_model.pth => models/FashionLabeling_model.pth | 0 quickStart/FashionLabeler.py | 2 +- quickStart/TrainFashionLabeler.py | 2 +- 3 files changed, 2 insertions(+), 2 deletions(-) rename FashionLabeling_model.pth => models/FashionLabeling_model.pth (100%) 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")