Upload modeling_moment.py

modeling_moment.py

@@ -366,6 +366,7 @@ class MomentEmbeddingModel(MomentPreTrainedModel):
         self.config = config
         self.seq_len = config.seq_len
         self.patch_len = config.patch_len
+        self.patch_stride_len = config.patch_stride_len
 
         # TODO: normalizer, tokenizerはProcessor側に配置するべきか？
         # 現状の考え: 特にMomentから切り離す用途もない。
@@ -515,7 +516,7 @@ class MomentEmbeddingModel(MomentPreTrainedModel):
         Returns:
             int: 計算されたn_patchesの数
         """
-        stride = self.stride if self.stride is not None else self.patch_len
+        stride = self.patch_stride_len if self.patch_stride_len is not None else self.patch_len
         n_patches = (seq_len - self.patch_len) // stride + 1
         return n_patches