Best Performing Model

.gitattributes

@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 generator_model_25_epochs_at_2023-04-16-18_12.pth filter=lfs diff=lfs merge=lfs -text
 generator_model_16of50_epochs_at_2023-04-16-23_53.pth filter=lfs diff=lfs merge=lfs -text
+generator_model_epoch_94.pth filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -1,3 +1,4 @@
 log.csv
 flagged
-__pycache__
+__pycache__
+.DS_Store

app.py

@@ -4,7 +4,7 @@ from utils import load_model, generate_random_img
 
 def generate_image():
     with torch.no_grad():
-        model = load_model('generator', 'generator_model_16of50_epochs_at_2023-04-16-23_53.pth')
+        model = load_model('generator', 'generator_model_epoch_94.pth')
         generated_image = generate_random_img(model)
         return generated_image
 

generator_model_16of50_epochs_at_2023-04-16-23_53.pth → generator_model_epoch_94.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8126c99c84c96e0946b8abdbca511a0a369a52956d8abd923e365050b3d2f195
-size 51311597
+oid sha256:16fd332be2bed132c3c42fcdac18e4bbcf0d2ed94de28f62b41efbc4b4dab7ba
+size 51311551

utils.py

@@ -143,5 +143,5 @@ def generate_random_img(model):
   return generated_image
 
 if __name__ == "__main__":
-    model = load_model('generator','generator_model_25_epochs_at_2023-04-16-18_12.pth')
+    model = load_model('generator','generator_model_epoch_94.pth)
     generate_random_img(model)