error occurs when model predicted

[yzj19870824]

When I execute command —— !cd coper-master/CoPER_MINERVA/ && bash ./experiment.sh configs/umls.sh --inference 0 error occurs: Memory allocated before eval data loading: 1.02944768 HIIIII 652 triples loaded from data/umls/dev.triples 661 triples loaded from data/umls/test.triples Memory allocated after eval data loading: 1.02944768 0% 0/326 [00:00<?, ?it/s] Traceback (most recent call last): File "/usr/lib/python3.7/runpy.py", line 193, in _run_module_as_main "main", mod_spec) File "/usr/lib/python3.7/runpy.py", line 85, in _run_code exec(code, run_globals) File "/content/coper-master/CoPER_MINERVA/src/experiments.py", line 830, in run_experiment(args) File "/content/coper-master/CoPER_MINERVA/src/experiments.py", line 813, in run_experiment inference(lf) File "/content/coper-master/CoPER_MINERVA/src/experiments.py", line 369, in inference pred_scores = lf.forward(dev_data, verbose=False) File "/content/coper-master/CoPER_MINERVA/src/learn_framework.py", line 253, in forward pred_score = self.predict(mini_batch, verbose=verbose) File "/content/coper-master/CoPER_MINERVA/src/rl/graph_search/pg.py", line 226, in predict pn, e1, r, e2, kg, self.num_rollout_steps, self.beam_size) File "/content/coper-master/CoPER_MINERVA/src/rl/graph_search/beam_search.py", line 168, in beam_search pn.update_path(action, kg, offset=action_offset) File "/content/coper-master/CoPER_MINERVA/src/rl/graph_search/pn.py", line 257, in update_path offset_path_history(self.path, offset) File "/content/coper-master/CoPER_MINERVA/src/rl/graph_search/pn.py", line 235, in offset_path_history new_tuple = tuple([_x[offset, :, :] for _x in x]) File "/content/coper-master/CoPER_MINERVA/src/rl/graph_search/pn.py", line 235, in new_tuple = tuple([_x[offset, :, :] for _x in x]) IndexError: tensors used as indices must be long, byte or bool tensors

type of elements in offset is float, how can I solve it, anyone can help

[otiliastr]

Hi,

I will try to reproduce your error. Could you please tell me what system and version of Tensorflow you are using?

Thanks, Otilia