Created
February 24, 2020 23:19
-
-
Save dreiss/76171f2d9241e78f805ef48c11f4bbdf to your computer and use it in GitHub Desktop.
in my test, it doesn’t running on GPU in java.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
It exactly not work without setting all the parameters as requires_grad=False. But, please how to set this?