How to customize trainer in order to restrict parameter range during training?

Thanks a lot for the solution :slight_smile: