how to train my cifar-10-batches-py?

[Asuna88]

I mkdir 'cifar-10-batches-py' and copy the data "data_batch_*".
after that, I run the script "python watermarks.py --logdir=./save "
but I got this error:

''
Traceback (most recent call last):
File "watermarks.py", line 311, in
tf.app.run()
File "lib/python3.6/site-packages/tensorflow/python/platform/app.py", line 40, in run
_run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef)
File "lib/python3.6/site-packages/absl/app.py", line 303, in run
_run_main(main, args)
File "lib/python3.6/site-packages/absl/app.py", line 251, in _run_main
sys.exit(main(argv))
File "watermarks.py", line 306, in main
train(sess, globals()[FLAGS.dataset])
File "watermarks.py", line 189, in train
next_image, iterator_inits = dataset_split(dataset, .8)
File "cnn-watermark-removal/dataset.py", line 70, in dataset_split
records = get_records()
File "cnn-watermark-removal/dataset.py", line 64, in get_records
convert_to_record()
File "cnn-watermark-removal/dataset.py", line 181, in convert_to_record
writer.close()
AttributeError: 'NoneType' object has no attribute 'close'

''

1. So do you have any idea about this? Does the cifar-10-batches-py data need something mask like 'cat-selection.png'?
2. If I want to train with my own data(like flowers with watermark), so what should I prepare? Can you list them? just like (1)original img, (2)mask png (3)watermark img? like this?

thx a lot!