SNN2.src.model.callbacks.ControlledCrossEntropy_better module