Addresses blackgnezdo nitpicks related to comments.
This commit is contained in:
parent
03164c04e8
commit
550ed59ad7
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue