about training new custom data set

[Zhang_Junjun]

I have trained new data set it worked and successful and all imag is resize 960*540。

but when i try to train smaller dataset: every img is 896 * 544, it is always print error:slice index 262144 of dimension 0 out of bounds。and i changed even smaller image size: 512*320, it print same error。

somewhere should i need to do?

[MaybeShewill-CV]

@zjj321 Change the config here

bisenetv2-tensorflow/config/cityscapes/cityscapes_bisenetv2.yaml

Lines 80 to 83 in 710db86

 OHEM: 
     ENABLE: True 
     SCORE_THRESH: 0.65 
     MIN_SAMPLE_NUMS: 262144 

You may change it into 0.5 * 896 * 544 or 0.25 * 896 * 544

[Zhang_Junjun]

best wishs,thank you so much