Dengguty

在运行test模式，BertAbs模型时，遇到了RuntimeError: "index_select_out_cuda_impl" not implemented for 'Float'

1

请问有人遇到这个问题吗？ 在解码的时候出现的问题，select_indices应该是整数类型的，但是debug发现是float，将其select_indices转为long型后，运行就一直卡住了，不出测试结果 python train.py -task abs -mode test -test_from ../models/abs/model_step_8000.pt -batch_size 3000 -test_batch_size 500 -bert_data_path ../bert_data/bert.pt_data -log_file ../logs/test_abs -model_path ../models/abs -sep_optim true -use_interval true -visible_gpus 0 -max_pos 512 -max_length 200...

请问您是用gpu跑的吗？cuda版本是多少？