#python-3.x #tensorflow2.0
#python-3.x #tensorflow2.0
Вопрос:
Я сохраняю контрольную точку и менеджер моей предварительно обученной модели (в файле python с именем train), я открываю другой файл и хотел бы восстановить или загрузить сохраненную контрольную точку и менеджера, чтобы настроить мою модель, как показано ниже:
checkpoint = tf.train.Checkpoint(step=tf.Variable(1), optimizer=None)
manager = tf.train.CheckpointManager(checkpoint, pretrain_save_path, max_to_keep=3)
Я нахожу эту ошибку
ValueError: `Checkpoint` was expecting a trackable object (an object derived from `TrackableBase`), got None. If you believe this object should be trackable (i.e. it is part of the TensorFlow Python API and manages state), please open an issue.
Ответ №1:
tf.train.Checkpoint()
Получает kwargs
пункты, которые являются ключевыми объектами. И объект должен быть отслеживаемым объектом ( None
не является отслеживаемым объектом). Поэтому вам нужно только удалить , optimizer=None