Open ashahba opened 5 years ago
@jakeret this is basically just bringing #202 up to date with master
.
I also realized the issue with https://github.com/IntelAI/models/blob/v1.4.0/benchmarks/image_segmentation/tensorflow/unet/README.md was that I was using checkpoint_name=model.cpkt
not realizing that it's now checkpoint_name=model.ckpt
and I fixed our docs.
Thanks.
@mpjlu would you also please review and provide feedback if needed.
Thanks.
hi @ashahba , thank you for your contribution. I wasn't aware that this repo is being used in IntelAI benchmarks, nice.
I hadn't merged #202 because of two reasons
the thread handling should not be part of the PR as it has nothing to do with the dropout
In my understanding if we set e.g. keep_prop != 1
e.g. 0.5
it can't be changed for validation or prediction (where we don't want any regularization) as it is a fix part of the graph. Or am I missing something?
Thanks @jakeret That sounds great. In the meantime I'm unblocked right now but I keep my eyes open for the any activity on #202
When using the image
gcr.io/deeplearning-platform-release/tf-cpu.1-14
and while following this steps: https://github.com/IntelAI/models/blob/v1.4.0/benchmarks/image_segmentation/tensorflow/unet/README.md I get the following error: