Я пытаюсь исправить полученную ошибку травления и думаю (надеюсь) нашел решение здесь 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)
Я использую tenorflow, поэтому у меня проблема с реализацией этого кода в tenorflow.
Я попытался найти эквивалент файла torch.load в тензорном потоке, и ближе всего я нашел класс tf.train.Saver. Тем не менее, они не очень похожи на то, как вы их используете, поэтому у меня возникли проблемы с их заменой.
Именно это я и пытаюсь перевести в тензорный поток. Я думаю, что все должно работать одинаково в тензорном потоке, кроме последней строки, с которой я борюсь.