i want to freeze EmbeddingCollection mid training, but setting the learning rate to 0.0 does not save cuda memory and computing resources. is there a method similar to the following code. looking forward to your reply, thanks

for param in module.parameters():
 param.requires_grad = False