summaryrefslogtreecommitdiff
path: root/candle-transformers/src/models/convmixer.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-transformers/src/models/convmixer.rs')
-rw-r--r--candle-transformers/src/models/convmixer.rs6
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};