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

about learning rate #3

Closed
chencodeX opened this issue Jul 12, 2017 · 3 comments
Closed

about learning rate #3

chencodeX opened this issue Jul 12, 2017 · 3 comments

Comments

@chencodeX
Copy link

In the second step of training, I used a smaller batch size, does that mean I need to use a smaller learning rate than 0.001?

@abhaydoke09
Copy link
Owner

This has some interesting discussions about batch_size vs learning_rate.
NVIDIA/DIGITS#51

@chencodeX
Copy link
Author

Another problem, in the second step, when I use 224 * 224 * 3 size data to train, it will prompt me to be wrong:
Traceback (most recent call last):
File "bcnn_finetuning.py", line 374, in
val_loss += sess.run(loss, feed_dict={imgs: batch_val_x, target: batch_val_y})
File "/opt/anaconda2/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 767, in run
run_metadata_ptr)
File "/opt/anaconda2/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 965, in _run
feed_dict_string, options, run_metadata)
File "/opt/anaconda2/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 1015, in _do_run
target_list, options, run_metadata)
File "/opt/anaconda2/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 1035, in _do_call
raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.InvalidArgumentError: logits and labels must be same size: logits_size=[2,100] labels_size=[8,100]
[[Node: SoftmaxCrossEntropyWithLogits = SoftmaxCrossEntropyWithLogits[T=DT_FLOAT, _device="/job:localhost/replica:0/task:0/gpu:0"](Reshape_2, Reshape_3)]]

I'm sorry, I'm just a novice.

@abhaydoke09
Copy link
Owner

abhaydoke09 commented Jul 27, 2017

I think the problem is in the image size.
Look at this issue.
#5

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