Thank you for your help. I asked the developer and they confirmed that the model requires a lot of memory and time for training, so I do not think that there is an issue with the performance because of another pytorch version