Thanks for the reply again. I’m clear how to pass the arguments to nn.CrossEntropyLoss but im not clear how do i pass the masks and images.
What values should i return in my ‘get__item’. Can you help me out with this, im new to pytorch.