diff options
Diffstat (limited to 'candle-transformers/src/models/convmixer.rs')
-rw-r--r-- | candle-transformers/src/models/convmixer.rs | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/candle-transformers/src/models/convmixer.rs b/candle-transformers/src/models/convmixer.rs index e095f793..7f1b75eb 100644 --- a/candle-transformers/src/models/convmixer.rs +++ b/candle-transformers/src/models/convmixer.rs @@ -1,10 +1,10 @@ //! ConvMixer implementation. //! //! See "Patches Are All You Need?" by Trockman et al. 2022 -//! - [Arxiv](https://arxiv.org/abs/2201.09792) -//! - [Github](https://github.com/locuslab/convmixer) //! - +//! - 📝 [Arxiv](https://arxiv.org/abs/2201.09792) +//! - 💻 [Github](https://github.com/locuslab/convmixer) +//! use candle::Result; use candle_nn::{batch_norm, Conv2dConfig, Module, VarBuilder}; |