Skip to content

Latest commit

 

History

History
33 lines (24 loc) · 754 Bytes

README.md

File metadata and controls

33 lines (24 loc) · 754 Bytes
Raw

HypeTrainer (v0.1)

HypeTrainer is a trainer for PyTorch models.

Installation

pip install git+https://github.com/JJBT/HypeTrainer.git

Usage

from trainer.factory import Factory
from trainer.trainer import Trainer
import hydra
from omegaconf import DictConfig, OmegaConf

def run_train(cfg):
    factory = Factory(cfg)
    trainer = Trainer(cfg, factory)
    trainer.run_train()

# Config format in examples (or you can override trainer.factory.Factory)
@hydra.main(config_path='conf', config_name='config')
def run(cfg: DictConfig):
    cfg = OmegaConf.create(cfg)
    run_train(cfg)

Contributing

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.