지원되는 참조 모델

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

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

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

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

프레임워크	주 버전	모델 카테고리	참조 모델	지원되는 버전
TensorFlow	2.x	이미지 분류	ResNet-2.x, MNIST-2.x, EfficientNet-2.x	지원되는 TensorFlow 버전을 참조하세요.
		언어 모델링	Transformer-2.x, BERT-2.x	지원되는 TensorFlow 버전을 참조하세요.
		객체 감지	RetinaNet-2.x	지원되는 TensorFlow 버전을 참조하세요.
		이미지 세분화	Mask-RCNN-2.x, ShapeMask-2.x	지원되는 TensorFlow 버전을 참조하세요.
		추천 시스템	DLRM-2.x, DCN-2.x, NCF-2.x	지원되는 TensorFlow 버전을 참조하세요.
PyTorch	2.x	이미지 분류	ResNet-PyTorch	1.13, 2.0
		이미지 생성	Stable Diffusion,	1.13, 2.0
JAX	최신	대규모 언어 모델	MaxText	최신
JAX/FLAX	최신	이미지 분류	ResNet50	최신