diff options
Diffstat (limited to 'candle-transformers/src/models/efficientnet.rs')
| -rw-r--r-- | candle-transformers/src/models/efficientnet.rs | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/candle-transformers/src/models/efficientnet.rs b/candle-transformers/src/models/efficientnet.rs index f15c9c79..ecca2509 100644 --- a/candle-transformers/src/models/efficientnet.rs +++ b/candle-transformers/src/models/efficientnet.rs @@ -1,3 +1,8 @@ +//! Implementation of EfficientBert, an efficient variant of BERT for computer vision tasks. +//! +//! See: +//! - ["EfficientBERT: Progressively Searching Multilayer Perceptron Architectures for BERT"](https://arxiv.org/abs/2201.00462) +//! use candle::{Result, Tensor, D}; use candle_nn as nn; use nn::{Module, VarBuilder}; |