[Pytorch] Stop updating params of a layer of a model during back-propagation

for param in self.model.Traj_Dec.decoder_mlp.parameters():
    param.requires_grad = False