diff --git a/src/main/resources/templates/gluon/CNNSupervisedTrainer.ftl b/src/main/resources/templates/gluon/CNNSupervisedTrainer.ftl index 34951554ad820535b13edb214e1b4b98afb8eee9..bf67e888db8d9d4b657d56350712a7c9b84d810a 100644 --- a/src/main/resources/templates/gluon/CNNSupervisedTrainer.ftl +++ b/src/main/resources/templates/gluon/CNNSupervisedTrainer.ftl @@ -51,7 +51,7 @@ class SoftmaxCrossEntropyLossIgnoreIndices(gluon.loss.Loss): if self._sparse_label: loss = -pick(pred, label, axis=self._axis, keepdims=True) else: - label = _reshape_like(F, label, pred) + label = gluon.loss._reshape_like(F, label, pred) loss = -(pred * label).sum(axis=self._axis, keepdims=True) # ignore some indices for loss, e.g. tokens in NLP applications for i in self._ignore_indices: