FluxML
diff --git a/‎vision/vae_mnist/.gitignore
Lines changed: 1 addition & 0 deletions b/‎vision/vae_mnist/.gitignore
Lines changed: 1 addition & 0 deletions
diff --git a/‎vision/vae_mnist/Manifest.toml
Lines changed: 1047 additions & 537 deletions b/‎vision/vae_mnist/Manifest.toml
Lines changed: 1047 additions & 537 deletions
diff --git a/‎vision/vae_mnist/Project.toml
Lines changed: 7 additions & 4 deletions b/‎vision/vae_mnist/Project.toml
Lines changed: 7 additions & 4 deletions
diff --git a/‎vision/vae_mnist/output/clustering.png
118 KB b/‎vision/vae_mnist/output/clustering.png
118 KB
diff --git a/‎vision/vae_mnist/output/epoch_1.png
740 Bytes b/‎vision/vae_mnist/output/epoch_1.png
740 Bytes
diff --git a/‎vision/vae_mnist/output/epoch_10.png
-2.93 KB b/‎vision/vae_mnist/output/epoch_10.png
-2.93 KB
diff --git a/‎vision/vae_mnist/output/epoch_11.png
-2.83 KB b/‎vision/vae_mnist/output/epoch_11.png
-2.83 KB
diff --git a/‎vision/vae_mnist/output/epoch_12.png
-3.27 KB b/‎vision/vae_mnist/output/epoch_12.png
-3.27 KB
diff --git a/‎vision/vae_mnist/output/epoch_13.png
-3.77 KB b/‎vision/vae_mnist/output/epoch_13.png
-3.77 KB
diff --git a/‎vision/vae_mnist/output/epoch_14.png
-3.37 KB b/‎vision/vae_mnist/output/epoch_14.png
-3.37 KB
@@ -0,0 +1 @@
+*.jld2
@@ -1,18 +1,21 @@
 [deps]
-BSON = "fbb218c0-5317-5bc6-957e-2ee96dd4b1f0"
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 DrWatson = "634d3b9d-ee7a-5ddf-bec9-22491ea816e1"
 Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
 ImageIO = "82e4d734-157c-48bb-816b-45c225c6df19"
 ImageMagick = "6218d12a-5da1-5696-b52f-db25d2ecc6d1"
 Images = "916415d5-f1e6-5110-898d-aaa5f9f070e0"
+JLD2 = "033835bb-8acc-5ee8-8aae-3f567f8a3819"
 MLDatasets = "eb30cadb-4394-5ae3-aed4-317e484a6458"
+MLUtils = "f1d291b0-491e-4a28-83b9-f70985020b54"
+Optimisers = "3bd65402-5787-11e9-1adc-39752487f4e2"
 Plots = "91a5bcdd-55d7-5caf-9e0b-520d859cae80"
 ProgressMeter = "92933f4c-e287-5a05-a399-4b506db050ca"
 TensorBoardLogger = "899adc3e-224a-11e9-021f-63837185c80f"
+cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd"
 
 [compat]
-CUDA = "2.4, 3.12, 4"
-Flux = "0.13.9, 0.14"
+CUDA = "5"
+Flux = "0.14"
 MLDatasets = "0.7"
-julia = "1.6"
+julia = "1.9"