o
    *j                     @   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                       s<   e Zd Z fddZ fddZ fddZdd Z  ZS )	MovieSceneSegmentationTrainerc                       t  j|i | d S N)super__init__selfargskwargs	__class__ x/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/trainers/cv/movie_scene_segmentation_trainer.pyr
   
      z&MovieSceneSegmentationTrainer.__init__c                    r   r   )r	   trainr   r   r   r   r      r   z#MovieSceneSegmentationTrainer.trainc                    s   t  j|i |}|S r   )r	   evaluate)r   r   r   Zmetric_valuesr   r   r   r      s   z&MovieSceneSegmentationTrainer.evaluatec                 C   s   d S r   r   )r   modelZinputsr   r   r   prediction_step   s   z-MovieSceneSegmentationTrainer.prediction_step)__name__
__module____qualname__r
   r   r   r   __classcell__r   r   r   r   r      s
    r   N)	Zmodelscope.metainfor   Zmodelscope.trainers.builderr   Zmodelscope.trainers.trainerr   Zregister_moduleZmovie_scene_segmentationr   r   r   r   r   <module>   s
   