coreylowman · coreylowman · Oct 7, 2022 · Oct 7, 2022 · Oct 7, 2022
diff --git a/src/nn/module.rs b/src/nn/module.rs
@@ -1,5 +1,3 @@
-use crate::gradients::CanUpdateWithGradients;
-
 /// A unit of a neural network. Acts on the generic `Input`
 /// and produces `Module::Output`.
 ///
@@ -16,7 +14,7 @@ use crate::gradients::CanUpdateWithGradients;
 /// a Dropout layer, which uses an rng under the hood, may
 /// not modify the input tensor in [Module::forward()], since
 /// it cannot modify it's underlying rng.
-pub trait Module<Input>: ResetParams + CanUpdateWithGradients {
+pub trait Module<Input> {
     /// The type that this unit produces given `Input`.
     type Output;