지원되는 참조 모델
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 | 최신 |