Estimator Tensorflow предоставляет метод получения желаемых значений переменных после обучения/тестирования с использованием get_variable_value а>. Существуют ли аналогичные функции в Sagemaker Estimator, чтобы я мог получить веса после обучения моей модели.
Получите веса после тренировки от Sagemaker Estimator
Ответы (1)
Для объекта Estimator
в SageMaker Python SDK после вызова fit()
вы можете вызвать получение URL-адреса S3 ваших артефактов модели с помощью
model_artifacts_url = estimator.create_model().model_data
Сама модель сохраняется в корзине S3 в виде архива по этому адресу. Отсюда вы можете получить параметры модели из S3.
person
Leopd
schedule
10.07.2018
Я пытался использовать метод
get_variable_value
для получения весов из Estimator. Хотя это работает для оценщика тензорного потока локально, Оценщик Tensorflow Estimator от Sagemaker, похоже, не имеет метода get_variable_value
или get_variable_names
.
- person user10031054; 12.07.2018
Ааа, теперь я понимаю, что вы имеете в виду. Я обновлю свой ответ.
- person Leopd; 17.07.2018