우리는 더 많은 tf.estimator.Estimator 호환 코드를 기반으로 오래된 교육 코드를 번역하려고합니다. 초기 코드에서 대상 데이터 세트의 원본 모델을 미세 조정합니다. variables_to_restore 및 init_fnMonitoredTrainingSession의 조합을 사용하여 검사를 수행하기 전에 검사 점에서 일부 레이어 만로드됩니다. tf.estimator.Estimator 방식을 사용하면 이러한 종류의 가중치로드를 어떻게 얻을 수 있습니까? model_fn
tf.estimator.Estimator를 사용하여 체크 포인트 및 미세 조정로드하기
2 model_fn
가 EstimatorSpec
을 반환 당신에
1 사용 tf.train.init_from_checkpoint
: 당신은 두 가지 옵션이 있습니다
이 작품을 만들었습니까? 나는 똑같은 문제를 겪고있다. – GPhilo
해결책을 찾고자하는 사람들에게 문제는 해결되었습니다 (실제 예제 사용) : https://stackoverflow.com/q/47867748/3214872 – GPhilo