dhiraa / tener Goto Github PK

Traceback (most recent call last):
File "bin/trainer.py", line 189, in
app.run(main)
File "/home/ubuntu/anaconda3/envs/wjy/lib/python3.7/site-packages/absl/app.py", line 303, in run
_run_main(main, args)
File "/home/ubuntu/anaconda3/envs/wjy/lib/python3.7/site-packages/absl/app.py", line 251, in _run_main
sys.exit(main(argv))
File "bin/trainer.py", line 185, in main
trainer.train()
File "bin/trainer.py", line 148, in train
is_log=batch%100 == 0)
File "./src/tener/models/tener_transformer.py", line 114, in train_step
predictions, loss = self._loss(logits=logits, real=tar, is_training=is_training)
File "./src/tener/models/tener_transformer.py", line 81, in _loss
sequence_lengths=sequence_lengths)
File "/home/ubuntu/anaconda3/envs/wjy/lib/python3.7/site-packages/tensorflow_addons/text/crf.py", line 196, in crf_log_likelihood
transition_params)
File "/home/ubuntu/anaconda3/envs/wjy/lib/python3.7/site-packages/tensorflow_addons/text/crf.py", line 68, in crf_sequence_score
return _multi_seq_fn()
File "/home/ubuntu/anaconda3/envs/wjy/lib/python3.7/site-packages/tensorflow_addons/text/crf.py", line 61, in _multi_seq_fn
transition_params)
File "/home/ubuntu/anaconda3/envs/wjy/lib/python3.7/site-packages/tensorflow_addons/text/crf.py", line 264, in crf_binary_score
flattened_transition_indices)
File "/home/ubuntu/anaconda3/envs/wjy/lib/python3.7/site-packages/tensorflow_core/python/util/dispatch.py", line 180, in wrapper
return target(*args, **kwargs)
File "/home/ubuntu/anaconda3/envs/wjy/lib/python3.7/site-packages/tensorflow_core/python/ops/array_ops.py", line 3984, in gather_v2
batch_dims=batch_dims)
File "/home/ubuntu/anaconda3/envs/wjy/lib/python3.7/site-packages/tensorflow_core/python/util/dispatch.py", line 180, in wrapper
return target(*args, **kwargs)
File "/home/ubuntu/anaconda3/envs/wjy/lib/python3.7/site-packages/tensorflow_core/python/ops/array_ops.py", line 3967, in gather
return gen_array_ops.gather_v2(params, indices, axis, name=name)
File "/home/ubuntu/anaconda3/envs/wjy/lib/python3.7/site-packages/tensorflow_core/python/ops/gen_array_ops.py", line 4075, in gather_v2
_six.raise_from(_core._status_to_exception(e.code, message), None)
File "", line 3, in raise_from
tensorflow.python.framework.errors_impl.InvalidArgumentError: indices[7,4] = 146 is not in [0, 144) [Op:GatherV2]

Hi,
Is the code in this repo usable in its current form, i.e. - were you able to reproduce the results of the original paper using the code from this repo?

Thanks