class_docstr: Test ConvNeXt model
code: def test_model_names(self):
      try:
          create_model(model_name="wrong name")
      except ValueError:
          pass
  
      tiny_1k = create_model(model_name="convnext_tiny_1k")
      self.assertTrue(tiny_1k.depths == (3, 3, 9, 3))
      self.assertTrue(tiny_1k.dims == (96, 192, 384, 768))
      self.assertTrue(tiny_1k.num_classes == 1000)
  
      # tiny_22k = create_model(model_name='convnext_tiny_22k')
      # self.assertTrue(tiny_22k.depths == (3, 3, 9, 3))
      # self.assertTrue(tiny_22k.dims == (96, 192, 384, 768))
      # self.assertTrue(tiny_22k.num_classes == 21841)
  
      small_1k = create_model(model_name="convnext_small_1k")
      self.assertTrue(small_1k.depths == (3, 3, 27, 3))
      self.assertTrue(small_1k.dims == (96, 192, 384, 768))
      self.assertTrue(small_1k.num_classes == 1000)
  
      # small_22k = create_model(model_name='convnext_small_22k')
      # self.assertTrue(small_22k.depths == (3, 3, 27, 3))
      # self.assertTrue(small_22k.dims == (96, 192, 384, 768))
      # self.assertTrue(small_22k.num_classes == 21841)
  
      base_1k = create_model(model_name="convnext_base_1k")
      self.assertTrue(base_1k.depths == (3, 3, 27, 3))
      self.assertTrue(base_1k.dims == (128, 256, 512, 1024))
      self.assertTrue(base_1k.num_classes == 1000)
  
      # base_22k = create_model(model_name='convnext_base_22k')
      # self.assertTrue(base_22k.depths == (3, 3, 27, 3))
      # self.assertTrue(base_22k.dims == (128, 256, 512, 1024))
      # self.assertTrue(base_22k.num_classes == 21841)
  
      large_1k = create_model(model_name="convnext_large_1k")
      self.assertTrue(large_1k.depths == (3, 3, 27, 3))
      self.assertTrue(large_1k.dims == (192, 384, 768, 1536))
      self.assertTrue(large_1k.num_classes == 1000)
  
      # large_22k = create_model(model_name='convnext_large_22k')
      # self.assertTrue(large_22k.depths == (3, 3, 27, 3))
      # self.assertTrue(large_22k.dims == (192, 384, 768, 1536))
      # self.assertTrue(large_22k.num_classes == 21841)
  
      xlarge_22k = create_model(model_name="convnext_xlarge_22k")
      self.assertTrue(xlarge_22k.depths == (3, 3, 27, 3))
      self.assertTrue(xlarge_22k.dims == (256, 512, 1024, 2048))
      self.assertTrue(xlarge_22k.num_classes == 21841)
  
imports: import unittest
  import torch
  from towhee.models.convnext import create_model
label_desc: Use image node_id 2 for calling the TestModel obj's underlying member method code with example usage: obj.test_model_names() without return types