从不同的工作目录加载pytorch模型

Question

我想从保存所有文件的外部目录加载和访问预训练模型。目录结构如下：

-MyProject
----Model_checkpoint_and_scripts
------access_model.py
--run_model.py
--other files

当运行模型调用access_model.py时，它在当前工作目录中查找model.py但找不到它。正如建议here，我可以使用

the_model = TheModelClass(*args, **kwargs)
the_model.load_state_dict(torch.load(PATH))

但在这种情况下，保存所有参数以初始化模型的好方法是什么？我想腌制命令行args，但有一些像vocab大小这样的论据是计算出来的。

由于

Answer 1

我可以通过添加以下行来加载它：

here = os.path.dirname(os.path.abspath(__file__))
sys.path.append(here)