Traceback (most recent call last):


 "run_pretraining.py", line 496, in <module>

tf.app.run()


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/platform/app.py", line 40, in run

_run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef)


 "/usr/local/lib/python3.7/site-packages/absl/app.py", line 300, in run

_run_main(main, args)


 "/usr/local/lib/python3.7/site-packages/absl/app.py", line 251, in _run_main

sys.exit(main(argv))


 "run_pretraining.py", line 469, in main

estimator.train(input_fn=train_input_fn, max_steps=FLAGS.num_train_steps)


 "/usr/local/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/tpu/tpu_estimator.py", line 2876, in train

rendezvous.raise_errors()


 "/usr/local/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/tpu/error_handling.py", line 131, in raise_errors

six.reraise(typ, value, traceback)


 "/usr/local/lib/python3.7/site-packages/six.py", line 693, in reraise

raise value


 "/usr/local/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/tpu/tpu_estimator.py", line 2871, in train

saving_listeners=saving_listeners)


 "/usr/local/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py", line 367, in train

loss = self._train_model(input_fn, hooks, saving_listeners)


 "/usr/local/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py", line 1158, in _train_model

return self._train_model_default(input_fn, hooks, saving_listeners)


 "/usr/local/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py", line 1185, in _train_model_default

input_fn, ModeKeys.TRAIN))


 "/usr/local/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/estimator.py", line 1022, in _get_features_and_labels_from_input_fn

self._call_input_fn(input_fn, mode))


 "/usr/local/lib/python3.7/site-packages/tensorflow_estimator/python/estimator/tpu/tpu_estimator.py", line 2828, in _call_input_fn

return input_fn(**kwargs)


 "run_pretraining.py", line 371, in input_fn

cycle_length=cycle_length))


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/data/ops/dataset_ops.py", line 1853, in apply

return DatasetV1Adapter(super(DatasetV1, self).apply(transformation_func))


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/data/ops/dataset_ops.py", line 1290, in apply

dataset = transformation_func(self)


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/data/experimental/ops/interleave_ops.py", line 93, in _apply_fn

buffer_output_elements, prefetch_input_elements)


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/data/ops/readers.py", line 224, in __init__

map_func, self._transformation_name(), dataset=input_dataset)


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/data/ops/dataset_ops.py", line 2555, in __init__

self._function = wrapper_fn._get_concrete_function_internal()


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/eager/function.py", line 1355, in _get_concrete_function_internal

*args, **kwargs)


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/eager/function.py", line 1349, in _get_concrete_function_internal_garbage_collected

graph_function, _, _ = self._maybe_define_function(args, kwargs)


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/eager/function.py", line 1652, in _maybe_define_function

graph_function = self._create_graph_function(args, kwargs)


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/eager/function.py", line 1545, in _create_graph_function

capture_by_value=self._capture_by_value),


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/framework/func_graph.py", line 715, in func_graph_from_py_func

func_outputs = python_func(*func_args, **func_kwargs)


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/data/ops/dataset_ops.py", line 2549, in wrapper_fn

ret = _wrapper_helper(*args)


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/data/ops/dataset_ops.py", line 2489, in _wrapper_helper

ret = func(*nested_args)


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/data/ops/readers.py", line 335, in __init__

filenames, compression_type, buffer_size, num_parallel_reads)


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/data/ops/readers.py", line 295, in __init__

filenames = _create_or_validate_filenames_dataset(filenames)


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/data/ops/readers.py", line 57, in _create_or_validate_filenames_dataset

filenames = ops.convert_to_tensor(filenames, dtype=dtypes.string)


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/framework/ops.py", line 1087, in convert_to_tensor

return convert_to_tensor_v2(value, dtype, preferred_dtype, name)


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/framework/ops.py", line 1145, in convert_to_tensor_v2

as_ref=False)


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/framework/ops.py", line 1224, in internal_convert_to_tensor

ret = conversion_func(value, dtype=dtype, name=name, as_ref=as_ref)


 "/usr/local/lib/python3.7/site-packages/tensorflow/python/framework/ops.py", line 1018, in _TensorTensorConversionFunction

(dtype.name, t.dtype.name, str(t)))
ValueError: Tensor conversion requested dtype string for Tensor with dtype float32: 'Tensor("args_0:0", shape=(), dtype=float32)'

Traceback (most recent call last):


 "/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/tensor_shape.py", line 670, in merge_with

new_dims.append(dim.merge_with(other[i]))


 "/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/tensor_shape.py", line 143, in merge_with

self.assert_is_compatible_with(other)


 "/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/tensor_shape.py", line 116, in assert_is_compatible_with

other))
ValueError: Dimensions 1 and 512 are not compatible
