vits_model_1.pth => train loss: 0.442 && valid loss: 0.556