o
    ॵi0                     @   sF   d dl mZ d dlmZ d dlmZ ejejdG dd deZdS )    )Trainers)TRAINERS)EpochBasedTrainer)module_namec                       sD   e Zd Z fddZdd Z fddZ fddZd	d
 Z  ZS ) ImageInstanceSegmentationTrainerc                       t  j|i | d S N)super__init__selfargskwargs	__class__ n/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/trainers/cv/image_instance_segmentation_trainer.pyr
   
      z)ImageInstanceSegmentationTrainer.__init__c                 C   s   |S r   r   )r   datar   r   r   
collate_fn   s   z+ImageInstanceSegmentationTrainer.collate_fnc                    r   r   )r	   trainr   r   r   r   r      r   z&ImageInstanceSegmentationTrainer.trainc                    s   t  j|i |}|S r   )r	   evaluate)r   r   r   metric_valuesr   r   r   r      s   z)ImageInstanceSegmentationTrainer.evaluatec                 C   s   d S r   r   )r   modelinputsr   r   r   prediction_step   s   z0ImageInstanceSegmentationTrainer.prediction_step)	__name__
__module____qualname__r
   r   r   r   r   __classcell__r   r   r   r   r      s    r   N)	modelscope.metainfor   modelscope.trainers.builderr   modelscope.trainers.trainerr   register_moduleimage_instance_segmentationr   r   r   r   r   <module>   s
   