i meet ValueError

[newedward]

Hi , thanks for your code.But i meet some troubles when i run it.
i meet ValueError when i run CUDA_VISIBLE_DEVICES=3 python3 realmix.py --filters=32 --dataset=cifar10_aug50.1@250-500 --w_match=75 --beta=0.75 --custom_dataset=True --augment=cifar10 --tsa=linear_schedule

Traceback (most recent call last):
  File "realmix.py", line 389, in <module>
    app.run(main)
  File "/home/njuedu/anaconda3/envs/NJUEDU/lib/python3.7/site-packages/absl/app.py", line 312, in run
    _run_main(main, args)
  File "/home/njuedu/anaconda3/envs/NJUEDU/lib/python3.7/site-packages/absl/app.py", line 258, in _run_main
    sys.exit(main(argv))
  File "realmix.py", line 314, in main
    dataset = get_dataset()
  File "realmix.py", line 302, in get_dataset
    dataset = DATASETS[FLAGS.dataset]()
  File "/data/huawei_demo/tensorflow/realmix/realmix-master/libml/data.py", line 185, in create
    train_labeled=fn(train_labeled).map(augment[0], para),
  File "/data/huawei_demo/tensorflow/realmix/realmix-master/libml/data.py", line 69, in memoize
    dataset = dataset.prefetch(16)
  File "/home/njuedu/anaconda3/envs/NJUEDU/lib/python3.7/site-packages/tensorflow_core/python/data/ops/dataset_ops.py", line 1859, in prefetch
    return DatasetV1Adapter(super(DatasetV1, self).prefetch(buffer_size))
  File "/home/njuedu/anaconda3/envs/NJUEDU/lib/python3.7/site-packages/tensorflow_core/python/data/ops/dataset_ops.py", line 795, in prefetch
    return PrefetchDataset(self, buffer_size)
  File "/home/njuedu/anaconda3/envs/NJUEDU/lib/python3.7/site-packages/tensorflow_core/python/data/ops/dataset_ops.py", line 3665, in __init__
    **self._flat_structure)
  File "/home/njuedu/anaconda3/envs/NJUEDU/lib/python3.7/site-packages/tensorflow_core/python/ops/gen_dataset_ops.py", line 4881, in prefetch_dataset
    legacy_autotune=legacy_autotune, name=name)
  File "/home/njuedu/anaconda3/envs/NJUEDU/lib/python3.7/site-packages/tensorflow_core/python/framework/op_def_library.py", line 367, in _apply_op_helper
    g = ops._get_graph_from_inputs(_Flatten(keywords.values()))
  File "/home/njuedu/anaconda3/envs/NJUEDU/lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py", line 5979, in _get_graph_from_inputs
    _assert_same_graph(original_graph_element, graph_element)
  File "/home/njuedu/anaconda3/envs/NJUEDU/lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py", line 5914, in _assert_same_graph
    (item, original_item))
ValueError: Tensor("buffer_size:0", shape=(), dtype=int64) must be from the same graph as Tensor("ParallelMapDataset:0", shape=(), dtype=variant).