@@ -135,19 +135,19 @@ mnist' = GanOpts <$> argument str (metavar "TRAIN")
136main :: IO ()
136main :: IO ()
138− GanOpts mnist iter rate load save <- execParser (info (mnist' <**> helper) idm)
138+ GanOpts mnist iter rate load save <- execParser (info (mnist' <**> helper) idm)
139− putStrLn "Training stupidly simply GAN"
139+ putStrLn "Training stupidly simply GAN"
140− nets0 <- case load of
140+ nets0 <- case load of
141− Just loadFile -> netLoad loadFile
141+ Just loadFile -> netLoad loadFile
142− Nothing -> (,) <$> randomDiscriminator <*> randomGenerator
142+ Nothing -> (,) <$> randomDiscriminator <*> randomGenerator
144− res <- runExceptT $ ganTest nets0 iter mnist rate
144+ res <- runExceptT $ ganTest nets0 iter mnist rate
145− case res of
145+ case res of
146− Right nets1 -> case save of
146+ Right nets1 -> case save of
147− Just saveFile -> B.writeFile saveFile $ runPut (put nets1)
147+ Just saveFile -> B.writeFile saveFile $ runPut (put nets1)
148− Nothing -> return ()
148+ Nothing -> return ()
150− Left err -> putStrLn err
150+ Left err -> putStrLn err
152readMNIST :: FilePath -> ExceptT String IO [(S ('D2 28 28), S ('D1 10))]
152readMNIST :: FilePath -> ExceptT String IO [(S ('D2 28 28), S ('D1 10))]
153readMNIST mnist = ExceptT $ do
153readMNIST mnist = ExceptT $ do