Feature Request: distinguish the parameters that require grad and that is not for PyTorch models