Feed Forward Pre-trained Inception in Batches
IdoWSC opened this issue · comments
Hi,
I am trying to use the Inception model used in the transfer learning tutorial. Specifically, I'm trying to use it with a batch size bigger then 1, to better utilize GPU's capabilities.
When looking at self.resized_image we get:
<tf.Tensor 'ResizeBilinear:0' shape=(1, 299, 299, 3) dtype=float32>
shape[0] = 1 means that the net will except only batch size of size 1.
Does anybody know if it can be modified to support a bigger batch size?
If it requires to bypass the image resizing its obviously also acceptable.
Thanks allot!
It has been so long since I did that tutorial that I don't remember. But I now generally recommend that you use Keras for doing Transfer Learning (see Tutorial 10) instead of the old and more direct ways in TensorFlow.