seems not save images when write_html

[CrossLee1]

In funciton write_html: _write_row(html_file, it, '%s/gen_train_current.jpg' % img_dir, all_size) It seems not save image named gen_train_current.jpg, so I cannot browse the html successfully