Error in turtlesim_semantic_pointcloud_example?

[crankler]

Hi, thank you for your good work.
When I launch roslaunch elevation_mapping_cupy turtlesim_semantic_pointcloud_example.launch, I found it can only show the color2points projection, and I try to project the semantic image to points.
However, I set semantic_segmentation: True and publish_segmentation_image: True, and use the initial lraspp_mobilenet_v3_large model. there happen errors:

[ERROR] [1710258357.603703, 2.620000]: bad callback: <bound method Subscriber.callback of <message_filters.Subscriber object at 0x7faaf82c4430>>
Traceback (most recent call last):
  File "/opt/ros/noetic/lib/python3/dist-packages/rospy/topics.py", line 750, in _invoke_callback
    cb(msg)
  File "/opt/ros/noetic/lib/python3/dist-packages/message_filters/__init__.py", line 76, in callback
    self.signalMessage(msg)
  File "/opt/ros/noetic/lib/python3/dist-packages/message_filters/__init__.py", line 58, in signalMessage
    cb(*(msg + args))
  File "/opt/ros/noetic/lib/python3/dist-packages/message_filters/__init__.py", line 330, in add
    self.signalMessage(*msgs)
  File "/opt/ros/noetic/lib/python3/dist-packages/message_filters/__init__.py", line 58, in signalMessage
    cb(*(msg + args))
  File "/home/yc/catkin_map/catkin_eth/src/elevation_mapping_cupy/sensor_processing/semantic_sensor/script/semantic_sensor/pointcloud_node.py", line 199, in image_callback
    pcl = self.create_pcl_from_image(image, depth, confidence)
  File "/home/yc/catkin_map/catkin_eth/src/elevation_mapping_cupy/sensor_processing/semantic_sensor/script/semantic_sensor/pointcloud_node.py", line 229, in create_pcl_from_image
    self.process_image(image, u, v, points)
  File "/home/yc/catkin_map/catkin_eth/src/elevation_mapping_cupy/sensor_processing/semantic_sensor/script/semantic_sensor/pointcloud_node.py", line 275, in process_image
    self.perform_segmentation(image, points, u, v)
  File "/home/yc/catkin_map/catkin_eth/src/elevation_mapping_cupy/sensor_processing/semantic_sensor/script/semantic_sensor/pointcloud_node.py", line 288, in perform_segmentation
    prediction = self.semantic_model["model"](image)
  File "/home/yc/catkin_map/catkin_eth/src/elevation_mapping_cupy/sensor_processing/semantic_sensor/script/semantic_sensor/networks.py", line 165, in __call__
    assert len(self.actual_channels) == selected_masks.shape[0]
AssertionError

[elevation_mapping-7] process has died [pid 34485, exit code -6, cmd /home/yc/catkin_map/catkin_eth/devel/lib/elevation_mapping_cupy/elevation_mapping_node __name:=elevation_mapping __log:=/home/yc/.ros/log/99e53228-e087-11ee-9c22-3b683d376962/elevation_mapping-7.log].
log file: /home/yc/.ros/log/99e53228-e087-11ee-9c22-3b683d376962/elevation_mapping-7*.log

Then I alos try the other 3 model, but they also show the same error: assert len(self.actual_channels) == selected_masks.shape[0].
Can you tell me how to fix this model error? Thank you for your help here~