Addresses blackgnezdo nitpicks related to comments.

This commit is contained in:
jcmartin 2020-11-05 17:17:23 +00:00
parent 03164c04e8
commit 550ed59ad7
2 changed files with 1 additions and 3 deletions

View File

@ -71,8 +71,6 @@ gradientDescent learningRate params grads = TF.withNameScope "gradientDescent" $
TF.assignAdd param (TF.scalar (-learningRate) `TF.mul` grad)
TF.group =<< zipWithM applyGrad params grads
-- TODO: Support more than Float in adam.
data AdamConfig t = AdamConfig
{ adamLearningRate :: t
, adamBeta1 :: t

View File

@ -104,7 +104,7 @@ testRereadRef = testCase "testReRunAssign" $ TF.runSession $ do
f1 <- TF.run w
liftIO $ (0.0, 0.1) @=? (TF.unScalar f0, TF.unScalar f1)
-- | Test einstein summation
-- | Test Einstein summation.
testEinsum :: Test
testEinsum = testCase "testEinsum" $ TF.runSession $ do
-- Matrix multiply