Works well on MNIST

This commit is contained in:
Joseph Redmon
2013-12-05 13:17:16 -08:00
parent 0d6bb5d44d
commit b715671988
12 changed files with 124 additions and 82 deletions

View File

@ -5,7 +5,7 @@
softmax_layer *make_softmax_layer(int inputs)
{
printf("Softmax Layer: %d inputs\n", inputs);
fprintf(stderr, "Softmax Layer: %d inputs\n", inputs);
softmax_layer *layer = calloc(1, sizeof(softmax_layer));
layer->inputs = inputs;
layer->output = calloc(inputs, sizeof(double));