yuanjun
Results
1
comments of
yuanjun
this is because the type of argument is wrong. you can change the type of argument by 'argument = argument.type(torch.FloatTensor)'