错误:Expected object of scalar type Long but got scalar type Int for argument #2 ‘target’ in call to _thnn_nll_loss_forward
代码更改如下
代码链接:MAML.
//源代码
accs = maml(x_spt, y_spt, x_qry, y_qry)
//更改后
accs = maml(x_spt, y_spt.long(
错误:Expected object of scalar type Long but got scalar type Int for argument #2 ‘target’ in call to _thnn_nll_loss_forward
代码更改如下
代码链接:MAML.
//源代码
accs = maml(x_spt, y_spt, x_qry, y_qry)
//更改后
accs = maml(x_spt, y_spt.long(