Hi @DAA233 Thanks for you great work. I have got the following error in torch.stack after completing the 3k iterations of training model. I unable to found the error.  Kindly help me out the error. Thanks in advance.