class_docstr: Test LightningDOT model
class_docstr_tok: ["Test","LightningDOT","model"]
code: def test_bi_encoder(self):
      batch = {
          "imgs": {
              "input_ids": 1,
              "attention_mask": 1,
              "position_ids": 1,
              "img_feat": 1,
              "img_pos_feat": 1,
              "img_masks": 1,
              "gather_index": 1,
          },
          "txts": {
              "input_ids": 1,
              "attention_mask": 1,
              "position_ids": 1,
              "img_feat": 1,
              "img_pos_feat": 1,
              "img_masks": 1,
              "gather_index": 1,
          },
          "caps": {
              "input_ids": 1,
              "attention_mask": 1,
              "position_ids": 1,
              "img_feat": 1,
              "img_pos_feat": 1,
              "img_masks": 1,
              "gather_index": 1,
          },
      }
      out = self.model(batch)
      self.assertTrue(out[0].requires_grad is True)
      self.assertTrue(out[1].requires_grad is True)
      self.assertTrue(out[2].requires_grad is True)
  
file_path: towhee/tests/unittests/models/lightning_dot/test_lightning_dot.py
imports: import unittest
  import torch
  from torch import nn
  from towhee.models.lightning_dot.bi_encoder import BiEncoder
label_desc: Use image node_id 1 for calling the TestLightningDOT obj's underlying member method code with example usage: obj.test_bi_encoder() without return types