trainer.py [76:83]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    args,
    model,
    optimizer,
    scheduler,
    X,
    Y,
    h_cache,
    stat,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



trainer.py [201:208]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            args,
            model,
            optimizer,
            scheduler,
            X,
            Y,
            h_cache,
            stat,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -