0

到目前为止,我一直在尝试获取 libtorch 中几个模块的哈希值,但无济于事。在 libtorch 中有几个与散列相关的函数,例如 :get_hashsimple_hash_gettorch/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)
4

0 回答 0