尝试在 PyTorch 中加载检查点时没有名为“parse_config”的模块答案

【问题标题】：No module named 'parse_config' while tryhing to load checkpoint in PyTorch尝试在 PyTorch 中加载检查点时没有名为“parse_config”的模块
【发布时间】：2020-10-26 00:38:56
【问题描述】：

在 Pytorch 中训练模型后，我保存了一个检查点文件。我必须在不同的模块中检查它，所以我尝试使用以下代码加载检查点。

map_location = lambda storage, loc: storage
checkpoint = torch.load("model.pt", map_location=map_location)

但它引发了ModuleNotFoundError 问题，我找不到解决方法。

错误回溯：

Traceback (most recent call last):
  File "main.py", line 11, in <module>
    model = loadmodel(hook_feature)
  File "/home/../model_loader.py", line 21, in loadmodel
    checkpoint = torch.load(settings.MODEL_FILE, map_location=map_location)
  File "/home/../.conda/envs/envreporting/lib/python3.6/site-packages/torch/serialization.py", line 584, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "/home/../.conda/envs/envreporting/lib/python3.6/site-packages/torch/serialization.py", line 842, in _load
    result = unpickler.load()
ModuleNotFoundError: No module named 'parse_config'

我找不到与此相关的现有问题。

【问题讨论】：

标签： python deep-learning pytorch

【解决方案1】：

您是否有可能使用https://github.com/victoresque/pytorch-template 来训练模型？在这种情况下，项目还将其配置保存在检查点中，您还需要导入 parse_config.py 文件才能加载它。

【讨论】：