-
Notifications
You must be signed in to change notification settings - Fork 900
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.
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
mnist demo not working #18
Comments
I pushed a "quick fix" by just adding back in |
Since we can load the complete |
Also, should we leave function buildModelGraph(checkpoints) {
var g = new deeplearn.Graph();
var input = g.placeholder('input', [784]);
var hidden1W = g.constant(checkpoints['hidden1/weights']);
var hidden1B = g.constant(checkpoints['hidden1/biases']);
var hidden1 = g.relu(g.add(g.matmul(input, hidden1W), hidden1B));
var hidden2W = g.constant(checkpoints['hidden2/weights']);
var hidden2B = g.constant(checkpoints['hidden2/biases']);
var hidden2 = g.relu(g.add(g.matmul(hidden1, hidden2W), hidden2B));
var softmaxW = g.constant(checkpoints['softmax_linear/weights']);
var softmaxB = g.constant(checkpoints['softmax_linear/biases']);
var logits = g.add(g.matmul(hidden2, softmaxW), softmaxB);
return [input, g.argmax(logits)];
} Do you think we should modify/simplify this? |
I'm not sure, but probably? I think we hold off for now as working through other examples and scenarios will inform our thinking. I could imagine a keras-like layer perhaps above deeplearn for architecting a model? We could also do a simple MNIST demo with the I also like the idea of avoiding MNIST and using something more artist-friendly for a classification demo as discussed in #8. |
This relates to the old examples and API. |
No description provided.
The text was updated successfully, but these errors were encountered: