tf.contrib.framework.assign_from_checkpoint_fn
Возвращает функцию, которая присваивает определенные переменные из контрольной точки.
tf.contrib.framework.assign_from_checkpoint_fn(
model_path, var_list, ignore_missing_vars=False, reshape_variables=False
)
Если ignore_missing_vars равно True и переменные не найдены в контрольной точке, возвращается None.
| Аргументы | |
|---|---|
model_path | Полный путь к контрольной точке модели. Чтобы получить последнюю контрольную точку, используйте model_path = tf.train.latest_checkpoint(checkpoint_dir) |
var_list | Список объектов Variable или словарь, сопоставляющий имена в контрольной точке соответствующим переменным для инициализации. Если пустой или None, возвращается no_op(), None. |
ignore_missing_vars | Булево значение. Если True, пропускаются переменные, отсутствующие в контрольной точке, с предупреждением вместо ошибки. |
reshape_variables | Булево значение. Если True, автоматически изменяет размер переменных, которые имеют другой размер, чем переменные в контрольной точке, но имеют то же количество элементов. |
| Возвращаемое значение | |
|---|---|
Функция, которая принимает один аргумент, a tf.compat.v1.Session, который применяет операцию присваивания. Если соответствующие переменные не были найдены в контрольной точке, возвращается None. |
| Исключения | |
|---|---|
ValueError | Если var_list пустой. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/framework/assign_from_checkpoint_fn