UM-CSCI-591-Spring-2020 · colligant · Mar 12, 2020
diff --git a/basic_gan.ipynb b/basic_gan.ipynb
diff --git a/data/mnist/MNIST/processed/test.pt b/data/mnist/MNIST/processed/test.pt
diff --git a/data/mnist/MNIST/processed/training.pt b/data/mnist/MNIST/processed/training.pt
diff --git a/data/mnist/MNIST/raw/t10k-images-idx3-ubyte b/data/mnist/MNIST/raw/t10k-images-idx3-ubyte
diff --git a/data/mnist/MNIST/raw/t10k-images-idx3-ubyte.gz b/data/mnist/MNIST/raw/t10k-images-idx3-ubyte.gz
diff --git a/data/mnist/MNIST/raw/t10k-labels-idx1-ubyte b/data/mnist/MNIST/raw/t10k-labels-idx1-ubyte
diff --git a/data/mnist/MNIST/raw/t10k-labels-idx1-ubyte.gz b/data/mnist/MNIST/raw/t10k-labels-idx1-ubyte.gz
diff --git a/data/mnist/MNIST/raw/train-images-idx3-ubyte b/data/mnist/MNIST/raw/train-images-idx3-ubyte
diff --git a/data/mnist/MNIST/raw/train-images-idx3-ubyte.gz b/data/mnist/MNIST/raw/train-images-idx3-ubyte.gz
diff --git a/data/mnist/MNIST/raw/train-labels-idx1-ubyte b/data/mnist/MNIST/raw/train-labels-idx1-ubyte
diff --git a/data/mnist/MNIST/raw/train-labels-idx1-ubyte.gz b/data/mnist/MNIST/raw/train-labels-idx1-ubyte.gz
diff --git a/little_gan.py b/little_gan.py
@@ -4,8 +4,9 @@
 
 
 # Target distribution samples Gamma distributed
-gamma = torch.distributions.Gamma(1, 3)
-X_true = gamma.sample((1000, 1))
+gamma1 = torch.distributions.Gamma(1, 3)
+gamma2 = torch.distributions.Gamma(3, 1)
+X_true = gamma1.sample((1000, 1)) + gamma2.sample((1000, 1))
 
 # Target distribution samples MoG distributed (Exhibits mode collapse)
 # X_0 = torch.randn(1000)
@@ -43,7 +44,7 @@ def forward(self, z):
 Z_0 = torch.randn(1000, 1)
 
 # Run for 25000 iterations
-n_iters = 2500
+n_iters = 25000
 
 # Number of samples per update step
 m = 64