main.py

from src.train import train

if __name__ == "__main__":
    train()