From 367c5c7582c8df45f8d9f1cddbd9919b3b8086e1 Mon Sep 17 00:00:00 2001
From: Steven Braun <steven.braun.mz@gmail.com>
Date: Thu, 18 Jan 2024 17:52:59 +0100
Subject: [PATCH] Update gen training params in readme

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 2aed640..c13b7c4 100644
--- a/README.md
+++ b/README.md
@@ -28,7 +28,7 @@ python main_pl.py dataset=mnist batch_size=128 epochs=100 dist=normal D=5 I=32 S
 Generative training on MNIST:
 
 ``` sh
-python main_pl.py dataset=mnist D=3 I=10 R=1 S=10 lr=0.1 dist=binomial epochs=10 batch_size=128
+python main_pl.py dataset=mnist D=5 I=16 R=10 S=16 lr=0.1 dist=binomial epochs=10 batch_size=128
 ```
 
 ![MNIST Samples]( ./res/mnist_samples.png )