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