이 섹션에는 tensorflow 패키지와 별개이지만 전용 하위 사이트 페이지가 없는 프로젝트 및 패키지에 대한 추가 API 참조 페이지 모음이 포함되어 있습니다.
TensorFlow 모델 저장소
TensorFlow 모델 리포지토리 는 최신(SOTA) 모델 구현을 제공합니다.
공식/프로젝트 디렉토리에는 TensorFlow의 고급 API를 사용하는 SOTA 모델 모음이 포함되어 있습니다. 최신 TensorFlow API를 사용하여 잘 유지 관리, 테스트 및 최신 상태로 유지하기 위한 것입니다.
이러한 모델을 빌드하고 훈련하는 데 사용되는 라이브러리 코드는 pip 패키지로 사용할 수 있습니다. 다음을 사용하여 설치할 수 있습니다.
$ pip install tensorflow-models-official # For the latest release
$ #or
$ pip install tf-models-nightly # For the nightly build
소스에서 패키지를 설치하려면 다음 지침 을 참조하십시오.
tensorflow-models-official pip 패키지에는 tensorflow_models 및 orbit 의 두 가지 최상위 모듈이 포함되어 있습니다. 다음을 사용하여 가져올 수 있습니다.
import tensorflow_models as tfm
import orbit
텐서플로우 모델
API 참조 .
tensorflow_models 모듈은 모델 구축 및 교육 구성을 처리합니다. 애플리케이션별 기능은 tfm.vision 및 tfm.nlp 에서 사용할 수 있습니다.
궤도
API 참조 .
orbit 모듈은 TensorFlow에서 맞춤형 훈련 루프 코드를 작성하기 위한 유연하고 가벼운 라이브러리를 정의합니다. Orbit은 작동하는 모델 유형에 대해 유연합니다. Orbit를 사용하여 Keras 모델을 훈련할 수 있지만(Keras의 Model.fit ), Keras를 전혀 사용할 필요는 없습니다. Orbit은 tf.distribute 와 원활하게 통합되며 다양한 장치 유형(CPU, GPU 및 TPU)에서 실행을 지원합니다.