diff options
Diffstat (limited to 'candle-transformers/src/models/convmixer.rs')
| -rw-r--r-- | candle-transformers/src/models/convmixer.rs | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/candle-transformers/src/models/convmixer.rs b/candle-transformers/src/models/convmixer.rs index f5abfa5d..e095f793 100644 --- a/candle-transformers/src/models/convmixer.rs +++ b/candle-transformers/src/models/convmixer.rs @@ -1,3 +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) +//! + use candle::Result; use candle_nn::{batch_norm, Conv2dConfig, Module, VarBuilder}; |