class_docstr: Test RepMLP utils
code: def test_replknet_stage(self):
      x = torch.rand(1, 5, 4, 4).to(self.device)
  
      layer1 = RepLKNetStage(
          channels=5,
          num_blocks=2,
          stage_lk_size=7,
          drop_rate=[0.0, 0.1],
          small_kernel=3,
          dw_ratio=1,
          ffn_ratio=4,
          small_kernel_merged=False,
          norm_intermediate_features=False,
      ).to(self.device)
      self.assertTrue(isinstance(layer1.norm, nn.Identity))
      outs1 = layer1(x)
      # print(outs1.shape)
      self.assertTrue(outs1.shape == (1, 5, 4, 4))
  
      layer2 = RepLKNetStage(
          channels=5,
          num_blocks=2,
          stage_lk_size=7,
          drop_rate=0.1,
          small_kernel=3,
          dw_ratio=1,
          ffn_ratio=4,
          small_kernel_merged=True,
          norm_intermediate_features=True,
      ).to(self.device)
      self.assertTrue(isinstance(layer2.norm, nn.BatchNorm2d))
      outs2 = layer2(x)
      # print(outs2.shape)
      self.assertTrue(outs2.shape == (1, 5, 4, 4))
  
file_path: towhee/tests/unittests/models/replknet/test_utils.py
imports: import unittest
  import torch
  from torch import nn
  from towhee.models.layers.conv_bn_activation import Conv2dBNActivation
  from towhee.models.replknet import fuse_bn, ConvFFN, ReparamLargeKernelConv, RepLKBlock, RepLKNetStage
label_desc: Use image node_id 5 for calling the TestUtils obj's underlying member method code with example usage: obj.test_replknet_stage() without return types