mapleTan1113

关于 the L73-L83 in model/layer.py 中learnable tensor的问题

2

你好！ 在使用 ``` # learnable DCT init self.register_parameter('weight', self.get_dct_filter(height, width, mapper_x, mapper_y, channel)) # learnable random init self.register_parameter('weight', torch.rand(channel, height, width)) ``` 这两种初始化方法时，会出现以下bug： TypeError: cannot assign 'torch.FloatTensor' object to parameter...