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!