pyTorch:如何重置其状态?

Posted

技术标签:

【中文标题】pyTorch:如何重置其状态?【英文标题】:pyTorch: how to reset its state? 【发布时间】:2021-07-20 16:18:31 【问题描述】:

我正在使用 pyTorch,但出于某种原因 - 在重构我的代码之后 - 它不再工作了。错误是:

  File ".\api\deepmatcher\data\dataset.py", line 420, in load_cache
    cached_data = torch.load(cachefile)
  File "c:\...\lib\site-packages\torch\serialization.py", line 592, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "c:\...\lib\site-packages\torch\serialization.py", line 851, in _load
    result = unpickler.load()
ModuleNotFoundError: No module named 'deepmatcher'

经过一些研究,我了解到 pyTorch 以某种方式保存其状态 (state_dict?)、它使用的类等等。

如何将 pyTorch 重置为其普通状态,以便再次运行我的代码?也许删除我电脑上的一些文件(它是 Win10 顺便说一句)?或者运行一些神奇的 python 命令?

【问题讨论】:

【参考方案1】:

解决方案是从项目中删除所有cacheddata.pth 文件。

【讨论】:

以上是关于pyTorch:如何重置其状态?的主要内容,如果未能解决你的问题,请参考以下文章

如何在pytorch中获得双向2层GRU的最终隐藏状态

PyTorch笔记 - GRU(Gated Recurrent Unit)

PyTorch笔记 - GRU(Gated Recurrent Unit)网络结构

PyTorch笔记 - GRU(Gated Recurrent Unit)网络结构

如何在pytorch中展平张量?

pytorch中的parametervariable和buffer