2017-09-26 2 views
3

우리는 더 많은 tf.estimator.Estimator 호환 코드를 기반으로 오래된 교육 코드를 번역하려고합니다. 초기 코드에서 대상 데이터 세트의 원본 모델을 미세 조정합니다. variables_to_restoreinit_fnMonitoredTrainingSession의 조합을 사용하여 검사를 수행하기 전에 검사 점에서 일부 레이어 만로드됩니다. tf.estimator.Estimator 방식을 사용하면 이러한 종류의 가중치로드를 어떻게 얻을 수 있습니까? model_fntf.estimator.Estimator를 사용하여 체크 포인트 및 미세 조정로드하기

2 model_fnEstimatorSpec을 반환 당신에

1 사용 tf.train.init_from_checkpoint : 당신은 두 가지 옵션이 있습니다

+0

이 작품을 만들었습니까? 나는 똑같은 문제를 겪고있다. – GPhilo

+0

해결책을 찾고자하는 사람들에게 문제는 해결되었습니다 (실제 예제 사용) : https://stackoverflow.com/q/47867748/3214872 – GPhilo

답변

1

은, 첫 번째는 간단합니다. 비계는 EstimatorSpec을 통해 설정할 수 있습니다.

+0

복원 할 변수 목록을 'init_from_checkpoint'로 지정하는 방법 체크 포인트? 볼 수 있듯이 범위 이름을 맵 키/값으로 만 전달할 수 있습니다. – GPhilo