HypeTrainer is a trainer for PyTorch models.
pip install git+https://github.com/JJBT/HypeTrainer.gitfrom 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)Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.