HI~Can i ask about this error?

[CHIJUI0128]

i didnt change any word in model.py

Traceback (most recent call last):
File "main.py", line 44, in
main()
File "main.py", line 38, in main
FLAGS.batch_size, FLAGS.epochs, FLAGS.learning_rate)
File "/home/user/CHI_FCN/FC-DenseNet-TensorFlow-master/model.py", line 347, in train
total_train_cost / train_step, total_val_cost / val_step))
UnboundLocalError: local variable 'train_step' referenced before assignment

[HasnainRaz]

It seems there is something wrong with your training data. Number of images is incorrect. Please make sure your images are correctly being read and their paths are correct.