New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Both GPU and CPU enable #10

Merged

bwerness merged 1 commit into aws-samples:master from goldmermaid:master

Sep 9, 2020

Contributor

goldmermaid commented Sep 9, 2020

Major modification:

initialize NN with defined context: net.initialize(init=init.Xavier(), ctx=context)
move validation dataset to defined context:
"X_val = nd.array(X_val).as_in_context(context)\n", "y_val = nd.array(y_val).as_in_context(context)"


          gpu enable

c6a6b36

bwerness approved these changes

View reviewed changes

Contributor

bwerness left a comment

A good improvement to allow for the use of CPU and GPU contexts.

bwerness merged commit cbf5cd7 into aws-samples:master

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment