U &c @stddlZddlmZmZddlmZddlmZmZm Z ddl m Z e e Zer\ddlZeGddde ZdS) N) dataclassfield)Tuple)cached_propertyis_tf_available tf_required)TrainingArgumentsc@seZdZUedddidZeed<edddidZeed<ed dd idZeed <ed dd idZ eed<edddidZ e ed<edddidZ e ed<edddidZeed<eeede fdddZeeddddZeee dddZd S) TFTrainingArgumentsZadamhelpzhName of a Tensorflow optimizer among "adadelta, adagrad, adam, adamax, ftrl, nadam, rmsprop, sgd, adamw")defaultmetadataoptimizer_nameztext-classificationzXType of task, one of "text-classification", "token-classification", "question-answering"modeZSparseCategoricalCrossentropyzgName of a Tensorflow loss. For the list see: https://www.tensorflow.org/api_docs/python/tf/keras/losses loss_nameNz Name of TPUtpu_namerzEnd learning rate for optimizerend_lriz Run an evaluation every X steps. eval_stepsFzIActivate the trace to record computation graphs and profiling informationdebugztf.distribute.Strategy)returncCstdtjd}|jr,tjjdd}nz(|jrFtjj |j}n tjj }Wnt k rld}YnX|rtj |tj j|tjj|}nXt|dkrtjjdd}ns