My model segments all points in one class

[Nhunguts]

Hi IsaacGuan @IsaacGuan Thank you for publishing your code. I have used your code to train my own data. My data also has two classes like yours. However, my model can't segment any positive point. It segmented all points in one class.
What should I check now? Thank you.

[BXESYR]

You can check the parameter Is_trained is True or False.

[Nhunguts]

You can check the parameter Is_trained is True or False.

Hi BXESYR, in the train.py file, "is_training=True" in "def train_one_epoch(epoch_number)" and "is_training=True" in "def eval_one_epoch(epoch_number). is this right? Or do you mean something else need to edit? Please help, thank you.