torch在init中定义参数的时候不能往gpu上放（.cuda）

会不存在于state_dict里面，从而无法通过model.parameters()找到并进行优化。