Библиотека обучения с подкреплением Rlib/Ray: невозможно перезагрузить агент !!! ["веса"] отсутствуют

Question

Я обучил агента со средой ray / rlib для задачи обучения с подкреплением. Проблема в том, что я не могу его перезагрузить. Когда я использую: agent.restore(контрольная точка), возникает ошибка:

self.set_weights(состояние ["вес"])KeyError: 'веса'

Спасибо за вашу помощь