Tom_ska:

My Shape is [1, 384, 384, 384] for Target and the same for model-output.


This is the issue, so please take another look at my previous post, which explains the expected shapes.
The issue is that the model output should contain a class dimension in dim1, while the target should not.