code: def test_classnetmodel_shapes(model):
      model = model(input_size=128, n_class=8)
      model.eval()
      output_batch = model(INPUT_BATCH)
      assert output_batch.size() == (BATCH_SIZE, 8)
  
file_path: pykale/tests/predict/test_class_domain_nets.py
imports: import pytest
  import torch
  from kale.predict.class_domain_nets import ClassNet, ClassNetSmallImage, ClassNetVideo, ClassNetVideoConv, DomainNetSmallImage, DomainNetVideo, SoftmaxNet
  from kale.utils.seed import set_seed
label_desc: Use image node_id 3 for calling a global function with example usage: test_classnetmodel_shapes(model) without return types