pplonski / keras2cpp

This is a bunch of code to port Keras neural network model into pure C++.
MIT License
679 stars 153 forks source link

Cant verify results #45

Open peacherwu opened 4 years ago

peacherwu commented 4 years ago

I am able to recover the result in C++ as in your comment of mnist_cnn_one_iteration.py

# on my pc I got: #[[ 0.03729606 0.00783805 0.06588034 0.21728528 0.01093729 0.34730983 # 0.01350389 0.02174525 0.26624694 0.01195715]]

Yet my python script yields a totally different prediction:

$ python mnist_cnn_one_iteration.py 
Using Theano backend.
X_train shape: (60000, 1, 28, 28)
60000 train samples
10000 test samples
Train on 60000 samples, validate on 10000 samples
Epoch 1/1
60000/60000 [==============================] - 20s - loss: 1.8239 - val_loss: 1.1273
Prediction on saved sample:
[[ 0.01161582  0.1554343   0.13277179  0.30984354  0.02740803  0.12834443
   0.00520513  0.02088989  0.13061689  0.07787018]]

My pc configs:

Pls help...