o zi!@sddlZddlmZddlmZddlmZddlm Z ddl m Z m Z ddl mZdd d ZdddZd dded dfddZdddZdddZdddZdS)N)PossibleUserWarning) TrainerFn)MisconfigurationException) is_overridden)rank_zero_deprecationrank_zero_warnis_param_in_hook_signaturetrainer pl.TrainerreturncCs|j}|jjdur td|jjtjkrt||t||n&|jjtjkr,t |dn|jjtj kr9t |dn |jjtj krEt |dt |t |dS)zChecks that the model is configured correctly before the run is started. Args: trainer: Lightning Trainer. Its `lightning_module` (the model) to check the configuration. NzNUnexpected: Trainer state fn must be set before validating loop configuration.valtestpredict)lightning_modulestatefn ValueErrorrFITTING%__verify_train_val_loop_configuration$__verify_manual_optimization_support VALIDATING __verify_eval_loop_configurationTESTING PREDICTING&__verify_configure_model_configuration"__warn_dataloader_iter_limitationsr modelre/home/ubuntu/.local/lib/python3.10/site-packages/pytorch_lightning/trainer/configuration_validator.py_verify_loop_configurationss       r!rpl.LightningModulecCstd|}|s tdtd|}|std|jjjj}td|}|r+|s+td|r5|s5tdtdt t |d drHt d t |j d t t |d dr[t d t |j ddS)N training_stepzNo `training_step()` method defined. Lightning `Trainer` expects as minimum a `training_step()`, `train_dataloader()` and `configure_optimizers()` to be defined.configure_optimizerszNo `configure_optimizers()` method defined. Lightning `Trainer` expects as minimum a `training_step()`, `train_dataloader()` and `configure_optimizers()` to be defined.validation_stepzRYou passed in a `val_dataloader` but have no `validation_step`. Skipping val loop.zPYou defined a `validation_step` but have no `val_dataloader`. Skipping val loop.categorytraining_epoch_endz>Support for `training_epoch_end` has been removed in v2.0.0. `z` implements this method. You can use the `on_train_epoch_end` hook instead. To access outputs, save them in-memory as instance attributes. You can find migration examples in https://github.com/Lightning-AI/lightning/pull/16520.validation_epoch_endz@Support for `validation_epoch_end` has been removed in v2.0.0. `z` implements this method. You can use the `on_validation_epoch_end` hook instead. To access outputs, save them in-memory as instance attributes. You can find migration examples in https://github.com/Lightning-AI/lightning/pull/16520.)rrfit_loop epoch_loopval_loop _data_source is_definedrrcallablegetattrNotImplementedErrortype__name__)r rhas_training_stephas_optimizershas_val_loader has_val_steprrr r1s8   rstagecCs|dkrdn|d}t||}|dkr,|jdurtd|s(td|s*tddSdS|sA|dkr4dn|}td |d |d |dkrGd nd }tt||drbtd|dt|jd|ddS)Nr r%_steprz6`predict_step` cannot be None to run `Trainer.predict`forwardz3`Trainer.predict` requires `forward` method to run.validatezNo `z#()` method defined to run `Trainer.z`.r)test_epoch_endz Support for `z` has been removed in v2.0.0. `z.` implements this method. You can use the `on_z` hook instead. To access outputs, save them in-memory as instance attributes. You can find migration examples in https://github.com/Lightning-AI/lightning/pull/16520.)r predict_steprr/r0r1r2r3)rr8 step_namehas_steptrainer_methodepoch_end_namerrr r\s&  rcCsP|jrdS|jdur|jdkrtd|jd|jdkr&td|jddS)NrzhAutomatic gradient clipping is not supported for manual optimization. Remove `Trainer(gradient_clip_val=z')` or switch to automatic optimization.zrAutomatic gradient accumulation is not supported for manual optimization. Remove `Trainer(accumulate_grad_batches=)automatic_optimizationgradient_clip_valraccumulate_grad_batchesrrrr rws rcCs6tdd|j|j|j|jfDrtdtddSdS)z&Check if `dataloader_iter is enabled`.css&|]}|durt|dddVqdS)Ndataloader_iterT)explicitr).0step_fnrrr s z5__warn_dataloader_iter_limitations..aYou are using the `dataloader_iter` step flavor. If you consume the iterator more than once per step, the `batch_idx` argument in any hook that takes it will not match with the batch index of the last batch consumed. This might have unforeseen effects on callbacks or code that expects to get the correct index. This will also not work well with gradient accumulation. This feature is very experimental and subject to change. Here be dragons.r&N)anyr#r%r= test_steprr)rrrr rs rcCsLtd|r$t|j}td|rtd|d|dtd|ddSdS)Nconfigure_sharded_modelconfigure_modelzBoth `z.configure_model`, and `zm.configure_sharded_model` are overridden. The latter is deprecated and it should be replaced with the former.zYou have overridden `z.configure_sharded_model` which is deprecated. Please override the `configure_model` hook instead. Instantiation with the newer hook will be created on the device right away and have the right data type depending on the precision setting in the Trainer.)rr2r3 RuntimeErrorr)rnamerrr rs    r)r r r N)r r rr"r N)rr"r N)pytorch_lightningpl#lightning_fabric.utilities.warningsr pytorch_lightning.trainer.statesr&pytorch_lightning.utilities.exceptionsr)pytorch_lightning.utilities.model_helpersr%pytorch_lightning.utilities.rank_zerorr+pytorch_lightning.utilities.signature_utilsr r!rstrrrrrrrrr s       +