到目前为止,我一直在尝试获取 libtorch 中几个模块的哈希值,但无济于事。在 libtorch 中有几个与散列相关的函数,例如 :get_hash
和simple_hash_get
在torch/csrc/utils/hash.h
.
我试图为我的 jit 跟踪模型获取哈希,如下所示:
torch::jit::Module myModel;
myModel = torch::jit::load("mymodel.jit");
std::string module_hash = torch::get_hash(mymodel);
但这不起作用。我试图获取字符串中的底层模型表示(即 using dump_to_str
)并将其传递给get_hash()
,但这也不起作用!如何在 libtorch 中使用此功能?
笔记
我需要找到一种方法,而不是使用它可能不是一个好的/可靠的选择,因为它每次都会创建不同的哈希:
auto model_hash = std::hash<std::string>{}(model_dumped_as_str)