Src Just main training script and sanity check notebook. Specific information about modules in their respective folders: model datasets utils ui