How do I change the batch_size?

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

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

MIT License

170 stars 47 forks source link

How do I change the batch_size? #18

Open DongwhanLee opened 3 years ago

DongwhanLee commented 3 years ago

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:

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.

parthdave1210 commented 3 years ago

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?

jinghanxu1999 commented 3 years ago

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