o i @s<dZddlmZddlmZedgd   d dd ZdS) zBasic loop for training.)errors) tf_exportztrain.basic_train_loop)v1NcCs|durg}|dur i}d}|rMz,d}||}|s,||g|Ri||rWdn1s6wYWn tjyHd}Ynw|sdSdS)a(Basic loop to train a model. Calls `train_step_fn` in a loop to train a model. The function is called as: ```python train_step_fn(session, *args, **kwargs) ``` It is passed a `tf.compat.v1.Session` in addition to `args` and `kwargs`. The function typically runs one training step in the session. Args: supervisor: `tf.compat.v1.train.Supervisor` to run the training services. train_step_fn: Callable to execute one training step. Called repeatedly as `train_step_fn(session, *args **kwargs)`. args: Optional positional arguments passed to `train_step_fn`. kwargs: Optional keyword arguments passed to `train_step_fn`. master: Master to use to create the training session. Defaults to `""` which causes the session to be created in the local process. NTF)managed_session should_stopr AbortedError) supervisor train_step_fnargskwargsmaster should_retrysessr\/mnt/sdb/aimis/docanh/lib/python3.10/site-packages/tensorflow/python/training/basic_loops.pybasic_train_loops&  r)NNr)__doc__tensorflow.python.frameworkr tensorflow.python.util.tf_exportrrrrrrs