primeqa.util.transformers_utils.optimizer_utils.TransformerOptimize#
- class primeqa.util.transformers_utils.optimizer_utils.TransformerOptimize(hypers: primeqa.util.transformers_utils.hypers_base.HypersBase, num_instances_to_train_over: int, model)#
Bases:
object
Collects standard steps to train transformer call step_loss after computing each loss
Methods
backward_on_loss
optimizer_step
should_continue
step_loss