Function build_trainer

program.py:119–132

(model, optimizer, loss, train_loader, val_loader, \
                  device, flags, global_state)

def build_trainer(model, optimizer, loss, train_loader, val_loader, \
                  device, flags, global_state):
    if flags.Global.algorithm in ['CRNN', 'FAN', 'GRCNN', 'DAN', 'SAR', 'SATRN']:
        trainer = TrainerRec(
            device=device,
            model=model,
            optimizer=optimizer,
            loss=loss,
            val_loader=val_loader,
            train_loader=train_loader,
            flags=flags,
            global_state=global_state
        )
    return trainer