Some errors

[RayYoh]

When I run the code, some errors occur.
boston_housing.py Line 39 It should be n_selected.append(save.selected.sum().cpu().numpy())
utils.py Line 38 n_selected.append(save.selected.sum().cpu().numpy())

[Louis Abraham]

Can you paste your error message? If you want to propose a fix, you can fork the repository and open a Pull Request.

[RayYoh]

It is not a serious problem.
The error means that the data is on cuda not on cpu.

[Louis Abraham]

thanks for your error report!