Only batches of spatial targets supported (non-empty 3D tensors) but got targets of size: : [1, 1, 256, 256]

Thank you for your comment. I am not sure, what actually executable code snippet!

I gave a post regarding this issue in here.

Let me know your thoughts.