nlp - 属性错误 : module 'dill._dill' has no attribute 'stack'

Question

使用这个 colab https://colab.research.google.com/drive/12LjJazBl7Gam0XBPy_y0CTOJZeZ34c2v?usp=sharing

我的 CUDA 版本:11.2

什么时候做

train_dataset = train_dataset.map(
    process_data_to_model_inputs,
    batched=True,
    batch_size=batch_size,
    remove_columns=["article", "abstract", "section_names"],
)

AttributeError: module 'dill._dill' has no attribute 'stack'

我试过了

pip install dill==0.3.4

但不工作

如何解决这个问题，思考!

Answer 1

我是 dill 的作者。 dill._dill.stack 不是公共(public)接口(interface)的一部分，并且不能保证在那里。它在 0.3.5(及更高版本)中被删除。它在 0.3.4 中可用。如果您已经安装了 0.3.5.1 或大于 0.3.4 的东西，您将需要 pip uninstall 或 force-reinstall 具体的包版本。

参见:Installing specific package version with pip