Traceback (most recent call last):


 "variational_autoencoder.py", line 65, in <module>

validation_data=(x_test, x_test))


 "/users/sourabh.d/.local/lib/python2.7/site-packages/keras/engine/training.py", line 1083, in fit

self._make_train_function()


 "/users/sourabh.d/.local/lib/python2.7/site-packages/keras/engine/training.py", line 696, in _make_train_function

self.total_loss)


 "/users/sourabh.d/.local/lib/python2.7/site-packages/keras/optimizers.py", line 213, in get_updates

new_a = self.rho * a + (1. - self.rho) * K.square(g)


 "/users/sourabh.d/.local/lib/python2.7/site-packages/keras/backend/tensorflow_backend.py", line 576, in square

return tf.square(x)


 "/users/sourabh.d/.local/lib/python2.7/site-packages/tensorflow/python/ops/math_ops.py", line 345, in square

return gen_math_ops.square(x, name=name)


 "/users/sourabh.d/.local/lib/python2.7/site-packages/tensorflow/python/ops/gen_math_ops.py", line 2228, in square

result = _op_def_lib.apply_op("Square", x=x, name=name)


 "/users/sourabh.d/.local/lib/python2.7/site-packages/tensorflow/python/framework/op_def_library.py", line 458, in apply_op

as_ref=input_arg.is_ref).dtype.name


 "/users/sourabh.d/.local/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 628, in convert_to_tensor

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


 "/users/sourabh.d/.local/lib/python2.7/site-packages/tensorflow/python/framework/constant_op.py", line 180, in _constant_tensor_conversion_function

return constant(v, dtype=dtype, name=name)


 "/users/sourabh.d/.local/lib/python2.7/site-packages/tensorflow/python/framework/constant_op.py", line 163, in constant

tensor_util.make_tensor_proto(value, dtype=dtype, shape=shape))


 "/users/sourabh.d/.local/lib/python2.7/site-packages/tensorflow/python/framework/tensor_util.py", line 346, in make_tensor_proto

raise ValueError("None values not supported.")
ValueError: None values not supported.

Traceback (most recent call last):


 "electra_small/run_finetuning.py", line 323, in <module>

main()


 "electra_small/run_finetuning.py", line 319, in main

args.model_name, args.data_dir, **hparams))


 "electra_small/run_finetuning.py", line 270, in run_finetuning

model_runner.train()


 "electra_small/run_finetuning.py", line 183, in train

input_fn=self._train_input_fn, max_steps=self.train_steps)


 "/usr/local/lib/python3.5/dist-packages/tensorflow_estimator/python/estimator/tpu/tpu_estimator.py", line 3035, in train

rendezvous.raise_errors()


 "/usr/local/lib/python3.5/dist-packages/tensorflow_estimator/python/estimator/tpu/error_handling.py", line 136, in raise_errors

six.reraise(typ, value, traceback)


 "/usr/local/lib/python3.5/dist-packages/six.py", line 703, in reraise

raise value


 "/usr/local/lib/python3.5/dist-packages/tensorflow_estimator/python/estimator/tpu/tpu_estimator.py", line 3030, in train

saving_listeners=saving_listeners)


 "/usr/local/lib/python3.5/dist-packages/tensorflow_estimator/python/estimator/estimator.py", line 370, in train

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


 "/usr/local/lib/python3.5/dist-packages/tensorflow_estimator/python/estimator/estimator.py", line 1161, in _train_model

return self._train_model_default(input_fn, hooks, saving_listeners)


 "/usr/local/lib/python3.5/dist-packages/tensorflow_estimator/python/estimator/estimator.py", line 1191, in _train_model_default

features, labels, ModeKeys.TRAIN, self.config)


 "/usr/local/lib/python3.5/dist-packages/tensorflow_estimator/python/estimator/tpu/tpu_estimator.py", line 2857, in _call_model_fn

config)


 "/usr/local/lib/python3.5/dist-packages/tensorflow_estimator/python/estimator/estimator.py", line 1149, in _call_model_fn

model_fn_results = self._model_fn(features=features, **kwargs)


 "/usr/local/lib/python3.5/dist-packages/tensorflow_estimator/python/estimator/tpu/tpu_estimator.py", line 3159, in _model_fn

_train_on_tpu_system(ctx, model_fn_wrapper, dequeue_fn))


 "/usr/local/lib/python3.5/dist-packages/tensorflow_estimator/python/estimator/tpu/tpu_estimator.py", line 3604, in _train_on_tpu_system

device_assignment=ctx.device_assignment)


 "/usr/local/lib/python3.5/dist-packages/tensorflow_core/python/tpu/tpu.py", line 1277, in split_compile_and_shard

name=name)


 "/usr/local/lib/python3.5/dist-packages/tensorflow_core/python/tpu/tpu.py", line 992, in split_compile_and_replicate

outputs = computation(*computation_inputs)


 "/usr/local/lib/python3.5/dist-packages/tensorflow_estimator/python/estimator/tpu/tpu_estimator.py", line 3589, in multi_tpu_train_steps_on_single_shard

inputs=[0, _INITIAL_LOSS])


 "/usr/local/lib/python3.5/dist-packages/tensorflow_core/python/tpu/training_loop.py", line 178, in while_loop

condition_wrapper, body_wrapper, inputs, name="", parallel_iterations=1)


 "/usr/local/lib/python3.5/dist-packages/tensorflow_core/python/ops/control_flow_ops.py", line 2753, in while_loop

return_same_structure)


 "/usr/local/lib/python3.5/dist-packages/tensorflow_core/python/ops/control_flow_ops.py", line 2245, in BuildLoop

pred, body, original_loop_vars, loop_vars, shape_invariants)


 "/usr/local/lib/python3.5/dist-packages/tensorflow_core/python/ops/control_flow_ops.py", line 2170, in _BuildLoop

body_result = body(*packed_vars_for_body)


 "/usr/local/lib/python3.5/dist-packages/tensorflow_core/python/tpu/training_loop.py", line 121, in body_wrapper

outputs = body(*(inputs + dequeue_ops))


 "/usr/local/lib/python3.5/dist-packages/tensorflow_estimator/python/estimator/tpu/tpu_estimator.py", line 3588, in <lambda>

lambda i, loss: [i + 1, single_tpu_train_step(i)],


 "/usr/local/lib/python3.5/dist-packages/tensorflow_estimator/python/estimator/tpu/tpu_estimator.py", line 1715, in train_step

self._call_model_fn(features, labels))


 "/usr/local/lib/python3.5/dist-packages/tensorflow_estimator/python/estimator/tpu/tpu_estimator.py", line 1994, in _call_model_fn

estimator_spec = self._model_fn(features=features, **kwargs)


 "electra_small/run_finetuning.py", line 96, in model_fn

tvars, init_checkpoint)


 "/home/etetteh/electra_small/model/modeling.py", line 351, in get_assignment_map_from_checkpoint

for x in tf.train.list_variables(init_checkpoint):


 "/usr/local/lib/python3.5/dist-packages/tensorflow_core/python/training/checkpoint_utils.py", line 97, in list_variables

reader = load_checkpoint(ckpt_dir_or_file)


 "/usr/local/lib/python3.5/dist-packages/tensorflow_core/python/training/checkpoint_utils.py", line 65, in load_checkpoint

"given directory %s" % ckpt_dir_or_file)
ValueError: Couldn't find 'checkpoint' file or checkpoints in given directory <model_dir>
