Training tweaks and data loader