Hi,
What is wrong with nn.Parameter ?
l get the following error :
TypeError: cannot assign ‘torch.cuda.FloatTensor’ as parameter ‘weight’ (torch.nn.Parameter or None expected)
when l do the following :
self.weight = torch.nn.Parameter(torch.FloatTensor(7,32,32),requires_grad=True).cuda()

def forward(self,x):
     
     x=torch.bmm(self.weight,x)

Thank you