Skip to content

Commit

Permalink
Removing Module super trait requirements (#223)
Browse files Browse the repository at this point in the history
* Removing Module super traits

* Removing unused import
  • Loading branch information
coreylowman committed Oct 7, 2022
1 parent eaad41a commit 041a2e2
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions src/nn/module.rs
Original file line number Diff line number Diff line change
@@ -1,5 +1,3 @@
use crate::gradients::CanUpdateWithGradients;

/// A unit of a neural network. Acts on the generic `Input`
/// and produces `Module::Output`.
///
Expand All @@ -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;

Expand Down

0 comments on commit 041a2e2

Please sign in to comment.