Error when running code of advpicker

[LittleMouseInCoding]

As shown in the figure, I use the same environment as you suggest, still meet the bug, do you have any solution?

[iofu728]

Hi @LittleMouseInCoding, it seems that the sp_data_ne is a empty list.
You can replace the https://github.com/microsoft/vert-papers/blob/master/papers/AdvPicker/get_xl_data.py#L254 to

sp_data_ne = TensorDataset(*[torch.stack(ii) if ii else torch.tensor([])  for ii in sp_data_ne])

or increase the args.threshold.

[LittleMouseInCoding]

I changed the code as you suggested, and still can not reproduce the reported result.