marinone94's picture
fix training script
4e5c598