o ॵi@sFddlmZddlmZddlmZejejdGdddeZdS))Trainers)TRAINERS)EpochBasedTrainer) module_namecs<eZdZfddZfddZfddZddZZS) MovieSceneSegmentationTrainerctj|i|dSN)super__init__selfargskwargs __class__k/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/trainers/cv/movie_scene_segmentation_trainer.pyr z&MovieSceneSegmentationTrainer.__init__crr)r trainr rrrr rz#MovieSceneSegmentationTrainer.traincstj|i|}|Sr)r evaluate)r r r metric_valuesrrrrsz&MovieSceneSegmentationTrainer.evaluatecCsdSrr)r modelinputsrrrprediction_stepsz-MovieSceneSegmentationTrainer.prediction_step)__name__ __module__ __qualname__r rrr __classcell__rrrrrs    rN) modelscope.metainformodelscope.trainers.builderrmodelscope.trainers.trainerrregister_modulemovie_scene_segmentationrrrrrs