diff options
Diffstat (limited to 'mnist/main.hs')
| -rw-r--r-- | mnist/main.hs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mnist/main.hs b/mnist/main.hs index 2810e34..02ba6f7 100644 --- a/mnist/main.hs +++ b/mnist/main.hs @@ -59,9 +59,9 @@ read_samples f1 f2 = do mainMNIST :: IO () mainMNIST =do - let ep = 20 -- number of epochs + let ep = 10 -- number of epochs let mbs = 10 -- mini-batch size - let lr = 2 -- learning rate + let lr = 3 -- learning rate let cap = 999999 -- cap number of training samples putStrLn "= Init =" |