finetune_alexnet_with_tensorflow
finetune_alexnet_with_tensorflow copied to clipboard
compatibility update, port data load code to tf.data, etc.
minor changes to make this repo be compatible with tf 1.10.
- port data load code to tf.data, https://github.com/tensorflow/tensorflow/blob/r1.4/tensorflow/contrib/data/README.md
- remove warning, "softmax_cross_entropy_with_logits deprecated", https://github.com/tensorflow/minigo/issues/37
For anyone not sure how to make the changes.
- Use
data = tf.data.Dataset
instead oftf.contrib.data
. Same goes to Iterator - Use
tf.nn.softmax_cross_entropy_with_logits_v2
instead oftf.nn.softmax_cross_entropy_with_logits
I'll look over this during the next week. Have some trouble with our newborn at the moment and I don't find much time to cross check the PR. But will come back to this, I promise