Repo:

Model Config:

Train Config: