dfdx icon indicating copy to clipboard operation
dfdx copied to clipboard

Update Dropout and DropoutOnIn to use Module::forward_mut

Open coreylowman opened this issue 1 year ago • 0 comments

They should have the following behavior:

  1. Module::forward() do nothing
  2. Module::forward_mut() call dropout (i.e. if there is a tape, then it will modify, otherwise it will do nothing)

coreylowman avatar Aug 20 '22 15:08 coreylowman