这个问题是对requires_grad=True的tensor进行修改导致的报错
例如:
b = torch.rand((f_num,1), requires_grad=True)*2-1
就会报错
可以改成:
b = torch.rand((f_num,1))*2-1
b.requires_grad=True
或者用with torch.no_grad():
with torch.no_grad():
b = torch.rand((f_num,1), requires_grad=True)*2-1