Hi, @ragavvenkatesan Do you have any update? I am also implementing a knowledge distillation-like training that uses `mx.symbol.Group` to get the intermediate activations and met the same error.
--- [Visit Topic](https://discuss.mxnet.apache.org/t/custom-loss-function-shape-mismatch/573/4) or reply to this email to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.mxnet.apache.org/email/unsubscribe/06e3f9bc9bff6ed22955d208c36856afdc9439bca38160818c81f095a5b6d90e).
