지원되는 참조 모델

Cloud TPU는 빠르고 정확한 학습에 최적화된 일련의 참조 모델을 제공합니다.

Cloud TPU에서는 다음과 같은 TensorFlow, PyTorch, JAX/FLAX의 주 프레임워크와 부 프레임워크 출시 버전을 지원합니다.

TensorFlow 출시 버전 번호가 출시 버전 2.5.0으로 변경되었습니다. 주 TensorFlow 출시 버전 번호는 '0'으로 끝나고 모든 패치 출시 버전 번호는 '0'보다 큰 숫자로 끝납니다. 예를 들어 TF 2.10.0은 주 출시 버전이고 TF 2.10.1은 부 출시 버전입니다.

지원되는 프레임워크의 최신 버전을 실행하려면 주 출시 버전에 패치 출시 버전이 있는지 확인합니다. 있는 경우 주 출시 버전이 아닌 지원되는 최신 패치 출시 버전을 실행할 수 있습니다.

프레임워크 주 버전 모델 카테고리 참조 모델 지원되는 버전
PyTorch 2.x 이미지 분류 ResNet-PyTorch 1.13, 2.0
JAX 최신 대규모 언어 모델 MaxText 최신
JAX/FLAX 최신 이미지 분류 ResNet50 최신