ModuleNotFoundError:没有名为“pegasus”的模块

Question

我想试着用飞马来总结这篇文章https://github.com/google-research/pegasus

最初的回购协议自述文件建议使用谷歌云计算引擎，但我使用的是协作笔记本。我是日本的一名英语教师，我希望我的学生能轻松地试用这个软件。他们可以体验机器学习和英语文章摘要

我遵循了这个指示https://github.com/google-research/pegasus/tree/f76b63c2886748f7f5c6c9fb547456d8c6002562#setup

这是我的colab笔记本。 https://colab.research.google.com/drive/1p95tZcjhfuCLYh23X3S_gZqRoWVhpIlE?usp=sharing

这是我在笔记本上的代码

%tensorflow_version 1.x

!git clone https://github.com/google-research/pegasus

!export PYTHONPATH=/content/pegasus

%pip install -r /content/pegasus/requirements.txt

!mkdir /content/pegasus/ckpt

!gsutil cp -r gs://pegasus_ckpt/ /content/pegasus/ckpt/

!python /content/pegasus/pegasus/bin/train.py --params=aeslc_transformer \
--param_overrides=vocab_filename=ckpt/pegasus_ckpt/c4.unigram.newline.10pct.96000.model \
--train_init_checkpoint=ckpt/pegasus_ckpt/model.ckpt-1500000 \
--model_dir=ckpt/pegasus_ckpt/aeslc

然后，我得到这个错误消息

Traceback (most recent call last):
  File "/content/pegasus/pegasus/bin/train.py", line 17, in <module>
    from pegasus.data import infeed
ModuleNotFoundError: No module named 'pegasus'

此错误消息表示python无法导入“pegasus”模块，但我使用!export PYTHONPATH=/content/pegasus此命令创建了python路径

你能给我一些建议吗

Answer 1

你不能用这个

!export PYTHONPATH=/content/pegasus

用这个代替

import os
os.environ['PYTHONPATH'] += ':/content/pegasus'

下面是可以正确运行的an example notebook

我不使用/content/pegasus/pegasus。我直接使用/content/pegasus，方法是将它与!npx degit ..一起安装