9i5SSKrSSKJr SSKJr SSKJr SSKJ r SSK J r SSK J r JrJrJrJrJrJrJr SSKJr SS KJr \ R2"\R4S 9"S S \ 55rg) N)Mapping) distributed)Trainers)TRAINERS)EpochBasedTrainer)DEFAULT_MODEL_REVISION ConfigFields ConfigKeysHubsModeKeys ModelFileTasks TrainerStages) to_device)func_receive_dict_inputs) module_namecR^\rSrSrU4SjrU4SjrU4SjrSrSrSr Sr U=r $) ImageInpaintingTrainerc&>[TU]"U0UD6 gN)super__init__selfargskwargs __class__s o/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/trainers/cv/image_inpainting_trainer.pyrImageInpaintingTrainer.__init__s $)&)c&>[TU]"U0UD6 gr)rtrainrs rr#ImageInpaintingTrainer.trains  t&v&r!c(>[TU]"U0UD6nU$r)revaluate)rrr metric_valuesrs rr&ImageInpaintingTrainer.evaluates($9&9 r!cgr)rmodelinputss rprediction_step&ImageInpaintingTrainer.prediction_steps r!c~UR[R5 SUlURR 5 [ URUR5GH9nUR[R5 [U5Hup4[X@R5nX@l X0l [ S5H]nUR[R5 URURXE5 UR[R 5 M_ U? U=R"S- sl[$R&UlUS-UR*:dM O UR[R,5 U=RS- slGM< UR[R.5 g)z:Training loop used by `EpochBasedTrainer.train()` rN) invoke_hookr before_run_epochr+r#range _max_epochsbefore_train_epoch enumeraterdevice data_batch _inner_iterbefore_train_iter train_stepafter_train_iter_iterr TRAIN_modeiters_per_epochafter_train_epoch after_run)r data_loader_ir:idxs r train_loop!ImageInpaintingTrainer.train_loop!s@ 112  t{{D$4$45A   ]== >!*;!7 &z;;? ",#$  8C$$]%D%DEOODJJ @$$]%C%CD$O a %^^ q5D000"8   ]<< = KK1 KK%6( 001r!c$UR5 [RUl[ U[ 5(a;[ UR5(d!URR"S0UDSU0D6nOURRX#S9n[ U[5(d [S5eSU;Ga2S/n[/5nUH:nURUR5Vs/sH oU;dM UPM sn5 M< 0n UHnURUS5n U cM[ R""5(al[ R$"5(aRU R&R)5n [ R*"U R-[ R."555 U RXR1505 M UR2RU 5 OUR2RUS5 X@lgs snf)a Perform a training step on a batch of inputs. Subclass and override to inject custom behavior. Args: model (`TorchModel`): The model to train. inputs (`Dict[str, Union[torch.Tensor, Any]]`): The inputs and targets of the model. The dictionary will be unpacked before being fed to the model. Most models expect the targets under the argument `labels`. Check your model's documentation for all accepted arguments. Return: `torch.Tensor`: The tensor with training loss on this batch. optimizer_idx)rLz$"model.forward()" must return a dictlog_varslossNr*)r#r r@rA isinstancerrforwardr+_do_stepdict TypeErrorsetupdatekeysgetdist is_availableis_initializeddataclone all_reducediv_get_world_sizeitem log_buffer train_outputs) rr+r,rHrbdefault_keys_pattern match_keyskey_pkeyrMvalues rr=!ImageInpaintingTrainer.train_step=s$  ^^ f  '? 'N'N!KK00M6MMM!KK000KM-..BC C ] *$*8 RJ-!!$1$6$6$8I$8SSLS$8IK.H!%))#t4$((**t/B/B/D/D % 0 0 2 43F3F3H(IJOOS**,$78 " OO " "8 , OO " "=#< =*Js & H 3H )r4r;rAr:rb) __name__ __module__ __qualname____firstlineno__rr#r&r-rIr=__static_attributes__ __classcell__)rs@rrrs&*' 282+2+r!r)timecollections.abcrtorchrrXmodelscope.metainformodelscope.trainers.builderrmodelscope.trainers.trainerrmodelscope.utils.constantrr r r r r rrmodelscope.utils.data_utilsrmodelscope.utils.file_utilsrregister_moduleimage_inpaintingrr*r!rrzs_ #%(09===2@ h&?&?@]+.]+A]+r!