比如想跑train.py,那么可以写一个train.sh,内容如下:
#!/usr/bin/env bash
PYTHONPATH="/path/to/mymodel":$PYTHONPATH \
python train.py
这个脚本的作用是在运行train.py时,将 /path/to/mymodel 这个路径添加到PYTHONPATH 变量中,这样的话import 模块时,也会到这个路径下去搜索可import 模块
可以在train.py中添加这一行代码:
sys.path.append('/path/to/mymodel')
能够实现一样的效果