hq-jiang / instance-segmentation-with-discriminative-loss-tensorflow

Tensorflow implementation of "Semantic Instance Segmentation with a Discriminative Loss Function"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How do I change the batch_size?

DongwhanLee opened this issue · comments

Hi, I am Dongwhan Lee from Korea.

I am impressed by your work and trying to follow yours.
However, I encountered the error like below:

image

That error message appears after I changed batch_size=1 to batch_size=16 in "training.py".
I looked into "datagenerator.py" too but failed to how to fix the error.

Is there any way to change the batch_size? If so, could you let me know?
Plus, I also wonder where that 1 in (1, 512, 512, 3) comes from.

Thank you in advance.

Hi, This is Parth Dave.

I am trying to run this program on my machine. Can you please give me the Requirements to run the code?

I also have this problem, have you solved it? And can you tell me how did you solved it ?think you