implement torch.clamp inside the train function. Parameter('k', clip=(min, max))
implement torch.clamp inside the train function.
Parameter('k', clip=(min, max))