Skip to content

Commit 99ea5ae

Browse files
committed
Fix dropout in Flow
1 parent 3ad9590 commit 99ea5ae

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/ConvNetSharp.Flow/Ops/DropoutGradient.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ public override Volume<T> Evaluate(Session<T> session)
3333

3434
var dropoutOutput = this._dropout.Evaluate(session);
3535
var dropoutInput = this._dropout.Parents[0].Evaluate(session);
36-
var dropoutInputGradient = this._dropout.Derivate.Evaluate(session);
36+
var dropoutOutputGradient = this._dropout.Derivate.Evaluate(session);
3737
var droupoutProb = this._dropout.DropoutProbability.Evaluate(session);
3838

3939
if (this.Result == null || !Equals(this._lastInputShape, dropoutInput.Shape))
@@ -44,7 +44,7 @@ public override Volume<T> Evaluate(Session<T> session)
4444
this.Result = BuilderInstance<T>.Volume.SameAs(dropoutInput.Shape);
4545
}
4646

47-
dropoutOutput.DoDropoutGradient(dropoutInput, this.Result, dropoutInputGradient, droupoutProb);
47+
dropoutOutput.DoDropoutGradient(dropoutInput, dropoutOutputGradient, this.Result, droupoutProb);
4848

4949
return base.Evaluate(session);
5050
}

0 commit comments

Comments
 (0)