if self.config.predict_gaps:
            weight = make_loss_weights(
                self.nb_classes, const.BAD_ID, self.config.gaps_bad_weight
            )
            self.xents[const.GAP_TAGS] = nn.CrossEntropyLoss(