在Tensorflow中等效于'torch.load'

Question

我正在尝试解决此酸洗错误，我想（希望）在此找到解决方案https://github.com/CSAILVision/places365/issues/25

soravux用户在pytorch中提供了解决方案：

from functools import partial
import pickle
pickle.load = partial(pickle.load, encoding="latin1")
pickle.Unpickler = partial(pickle.Unpickler, encoding="latin1")
model = torch.load(model_file, map_location=lambda storage, loc: storage, pickle_module=pickle)

我正在使用tensorflow，所以我遇到的问题是在tensorflow中实现此代码。

我尝试在tensorflow中搜索与torch.load等效的东西，而我找到的最接近的东西是tf.train.Saver类。但是，它们的用法似乎不太相似，因此在替换它们时遇到了麻烦。

这就是我要转换为张量流的内容。我认为除了最后一行我在努力之外，其他所有事情都应该在张量流中工作。