diff options
| author | Ivar Flakstad <69173633+ivarflakstad@users.noreply.github.com> | 2024-01-07 11:52:03 +0100 |
|---|---|---|
| committer | Ivar Flakstad <69173633+ivarflakstad@users.noreply.github.com> | 2024-01-07 11:52:03 +0100 |
| commit | 6ebe04327397ebf7c9400d68c43fede705f8ce75 (patch) | |
| tree | 4e701ef55370cb6226593019f9e6daaf8ddf73b4 /candle-transformers/src/models/efficientnet.rs | |
| parent | 6bf52b9fdf82ad775611e82924d73172660a605e (diff) | |
| parent | 84250bf52f58528cf59dca3b82effd9f07a13cc7 (diff) | |
| download | candle-6ebe04327397ebf7c9400d68c43fede705f8ce75.tar.gz candle-6ebe04327397ebf7c9400d68c43fede705f8ce75.tar.bz2 candle-6ebe04327397ebf7c9400d68c43fede705f8ce75.zip | |
Merge branch 'main' into ivarflakstad/metal-prng
Diffstat (limited to 'candle-transformers/src/models/efficientnet.rs')
| -rw-r--r-- | candle-transformers/src/models/efficientnet.rs | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/candle-transformers/src/models/efficientnet.rs b/candle-transformers/src/models/efficientnet.rs index ab51c76d..f15c9c79 100644 --- a/candle-transformers/src/models/efficientnet.rs +++ b/candle-transformers/src/models/efficientnet.rs @@ -169,8 +169,7 @@ impl ConvNormActivation { impl Module for ConvNormActivation { fn forward(&self, xs: &Tensor) -> Result<Tensor> { - let xs = self.conv2d.forward(xs)?; - let xs = self.bn2d.forward(&xs)?; + let xs = self.conv2d.forward(xs)?.apply_t(&self.bn2d, false)?; if self.activation { swish(&xs) } else { |