more config changes

jimexist · Feb 20, 2024 · 2bb41bf · 2bb41bf
1 parent 83f9965
commit 2bb41bf
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/src/detection/mod.rs b/src/detection/mod.rs
@@ -1,3 +1,5 @@
+//! Detection module, consisting of segformer implementation
+
 mod segformer;
 
 use crate::error::Result;

diff --git a/src/detection/segformer.rs b/src/detection/segformer.rs
@@ -1,3 +1,4 @@
+//! Segformer implementation
 use candle_core::{Module, ModuleT, Result, Tensor, D};
 use candle_nn::{
     conv2d, conv2d_no_bias, layer_norm, linear, Activation, Conv2d, Conv2dConfig, Linear,

diff --git a/src/recognition/mod.rs b/src/recognition/mod.rs
@@ -16,7 +16,6 @@ use swin_transformer::SwinModel;
 pub struct Config {
     encoder: SwinConfig,
     decoder: MBartConfig,
-    pad_token_id: i32,
 }
 
 #[derive(Debug, Clone)]