thunth

I guess that error is because of the TensorFlow version. This work was implemented using TensorFlow 1.x and maybe you are using version 2.x. If that is the case, changing...

I solved it by adding the following function to the class Logger in file 'utils.py' ``` def flush(self): pass ```

Thank you, @YTEP-ZHI, for your reply ^^ I would appreciate it if someone could explain why the GPU memory is not released after the training is stopped forcibly.