train.py

from allamo.configuration import AllamoConfiguration
from allamo.trainer.simple_trainer import SimpleTrainer

if __name__ == '__main__':
    config = AllamoConfiguration()
    trainer = SimpleTrainer(config)
    trainer.init_wandb()
    trainer.train()
    trainer.close()