Hi,
@gphilip



 gphilip:

What is the error you got when you tried to do this?


I got the following error:
cannot assign 'torch.cuda.FloatTensor' as parameter 'mask_weight' (torch.nn.Parameter or None expected)


One more question/obervation:
if I use the code self.mask_weight.data = self.mask_weight.data*temp, there is no error,
however, self.mask_weight will not be optimized during the training.
is there any good syntax to do this?