Above is the error I am getting. Unfortunately I am not able to reproduce the error in a small example. A similar issue for ConcatBackward is addressed here https://github.com/pytorch/pytorch/issues/2367 20 but I am unable to figure out the same for SetItemBackward.
Any help will be appreciated. Thanks