Suggestion of update a code sentence that will occuor error

[Jiaen529]

Hi,

1. In the data folder, dataset.py. I have had an error says:"ZeroDivisionError: float division by zero" when trying run your train.py with no changes made. Since in train.py line 168 total_time / total_step will have total step=0. How about you change your line 21 in dataset.py to my version -- if "seg" in path.lower(): -- this will give you a more robust model and no error on my test on your data and model.
2. in train.py line 68, I also need to change the shuffle to non true and then it runs good, not typically understand why, please explain more if have time
   Thanks

[pengyouwei]

I met the same questions.

[obiyoag]

Hi! Thx for your issue. You can make a PR to fix that.