RuntimeError: result type Float can’t be cast to the desired output type long
改为:
test_loss = criterion(y_pred, y_gt.float()).item()
RuntimeError: result type Float can’t be cast to the desired output type long
改为:
test_loss = criterion(y_pred, y_gt.float()).item()