Skip to content
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

No model.ckpt was saved under dierctory model/ #8

Open
bababadboy opened this issue Oct 4, 2018 · 1 comment
Open

No model.ckpt was saved under dierctory model/ #8

bababadboy opened this issue Oct 4, 2018 · 1 comment

Comments

@bababadboy
Copy link

I ran python mnist_cnn_train.py in terminal and the terminal returned

Optimization` Finished!
test accuracy for the stored model: 0.9932

Training logs are saved in "logs/train" , however, there is no trained model saved as "model/model01_99.61/model.ckpt" or in other directories. When I run python mnist_cnn_test.py --model-dir model/model01_99.61 --batch-size 5000 --use-ensemble False and it returns an error message:

NotFoundError (see above for traceback): Key fc3/BatchNorm/beta not found in checkpoint
	 [[Node: save/RestoreV2 = RestoreV2[dtypes=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT], _device="/job:localhost/replica:0/task:0/device:CPU:0"](_arg_save/Const_0_0, save/RestoreV2/tensor_names, save/RestoreV2/shape_and_slices)]]

Where is the problem?

@bababadboy bababadboy changed the title No model.ckpt was saved under model/ No model.ckpt was saved under dierctory model/ Oct 4, 2018
@liaoweiquan
Copy link

liaoweiquan commented Dec 1, 2018

If you use Windows, you should run "python mnist_cnn_test.py --model-dir /model/model01_99.61 --batch-size 5000 --use-ensemble True"
All you have to do is add a '/' before "model"
Then your code can run.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants