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