class_docstr: EncImg module
class_docstr_tok: ["EncImg","module"]
code: def __init__(self):
      super().__init__()
  
      self.swin = VideoSwinTransformer(
          patch_size=(2, 4, 4),
          patch_norm=True,
          depths=[2, 2, 18, 2],
          window_size=(8, 7, 7),
          num_heads=[3, 6, 12, 24],
          mlp_ratio=4.0,
          drop_path_rate=0.2,
          stride=(1, 4, 4),
      )
      # self.swin = SwinTransformer3D()
      # self.swin.load_state_dict(torch.load("./_snapshot/ckpt_video-swin.pt", map_location="cpu"))
  
      self.emb_cls = torch.nn.Parameter(
          0.02 * torch.randn(1, 1, 1, 768)
      )
      self.emb_pos = torch.nn.Parameter(
          0.02 * torch.randn(1, 1, 1 + 14**2, 768)
      )
      self.emb_len = torch.nn.Parameter(
          0.02 * torch.randn(1, 6, 1, 768)
      )
      self.norm = torch.nn.LayerNorm(768)
  
file_path: towhee/towhee/models/violet/violet.py
imports: import torch
  import os
  from towhee.models.video_swin_transformer.video_swin_transformer import VideoSwinTransformer
label_desc: Use image node_id 1 to create a new EncImg object from inherited base classes: torch.nn with example: obj = EncImg()