Modelos compatibles

Cloud TPU proporciona el siguiente conjunto de modelos compatibles que están optimizados para un entrenamiento rápido y preciso.

Versión de TensorFlow	Inicio de la compatibilidad	Finalización de la compatibilidad	Categoría del modelo	Modelos compatibles
TF 2.4.0	21 de diciembre de 2020	(Fecha de finalización aún no establecida)	Clasificación de imágenes	ResNet-2.x,MNIST-2.x,EfficientNet-2.x,
			Modelado de lenguajes	Transformer-2.x,BERT-2.x
			Detección de objetos	RetinaNet‑2.x
			Segmentación de imágenes	Mask-RCNN-2.x, ShapeMask-2.x
			Sistemas de recomendación	NCF-2.x
TF 2.3.1	1 de octubre de 2020	(Fecha de finalización aún no establecida)	Clasificación de imágenes	ResNet-2.x,MNIST-2.x,EfficientNet-2.x,
			Modelado de lenguajes	Transformer-2.x,BERT-2.x
			Detección de objetos	RetinaNet‑2.x
			Segmentación de imágenes	Mask-RCNN-2.x, ShapeMask-2.x
			Sistemas de recomendación	NCF-2.x
TF 2.3	28 de julio de 2020	(Fecha de finalización aún no establecida)	Clasificación de imágenes	ResNet-2.x,MNIST-2.x,EfficientNet-2.x,
			Modelado de lenguajes	Transformer-2.x,BERT-2.x
			Detección de objetos	RetinaNet‑2.x
			Segmentación de imágenes	Mask-RCNN-2.x, ShapeMask-2.x
			Sistemas de recomendación	NCF-2.x
TF 2.2.2	12 de enero de 2020	(Fecha de finalización aún no establecida)	Clasificación de imágenes	ResNet-2.x,MNIST-2.x,EfficientNet-2.x,
			Modelado de lenguajes	Transformer-2.x,BERT-2.x
			Detección de objetos	RetinaNet‑2.x
			Segmentación de imágenes	Mask-RCNN-2.x
TF 2.2.1	7 de octubre de 2020	(Fecha de finalización aún no establecida)	Clasificación de imágenes	ResNet-2.x,MNIST-2.x,EfficientNet-2.x,
			Modelado de lenguajes	Transformer-2.x,BERT-2.x
			Detección de objetos	RetinaNet‑2.x
			Segmentación de imágenes	Mask-RCNN-2.x
TF 2.2	8 de mayo de 2020	(Fecha de finalización aún no establecida)	Clasificación de imágenes	ResNet-2.x,MNIST-2.x,EfficientNet-2.x,
			Modelado de lenguajes	Transformer-2.x,BERT-2.x
			Detección de objetos	RetinaNet‑2.x
			Segmentación de imágenes	Mask-RCNN-2.x
TF 2.1.2	30 de octubre de 2020	(Fecha de finalización aún no establecida)	Clasificación de imágenes	ResNet-2.x,MNIST-2.x
			Modelado de lenguajes	Transformer-2.x,BERT-2.x
			Detección de objetos	RetinaNet‑2.x
TF 2.1.1	21 de mayo de 2020	(Fecha de finalización aún no establecida)	Clasificación de imágenes	ResNet-2.x,MNIST-2.x
			Modelado de lenguajes	Transformer-2.x,BERT-2.x
			Detección de objetos	RetinaNet‑2.x
TF 2.1	9 de enero de 2020	30 de junio de 2020	Clasificación de imágenes	ResNet-2.x,MNIST-2.x
			Modelado de lenguajes	Transformer-2.x,BERT-2.x
			Detección de objetos	RetinaNet‑2.x
TF 1.15.5	14 de enero de 2020	(Fecha de finalización aún no establecida)	Clasificación de imágenes	ResNet, AmoebaNet, EfficientNet, MNasNet, MNIST
			Modelado de lenguajes	Transformer, BERT
			Detección de objetos	RetinaNet
			Segmentación de imágenes	Mask RCNN, ShapeMask
TF 1.15.4	21 de octubre de 2020	(Fecha de finalización aún no establecida)	Clasificación de imágenes	ResNet, AmoebaNet, EfficientNet, MNasNet, MNIST
			Modelado de lenguajes	Transformer, BERT
			Detección de objetos	RetinaNet
			Segmentación de imágenes	Mask RCNN, ShapeMask
TF 1.15.3	29 de mayo de 2020	(Fecha de finalización aún no establecida)	Clasificación de imágenes	ResNet, AmoebaNet, EfficientNet, MNasNet, MNIST
			Modelado de lenguajes	Transformer, BERT
			Detección de objetos	RetinaNet
			Segmentación de imágenes	Mask RCNN, ShapeMask
TF 1.15.2	12 de mayo de 2020	30 de junio de 2020	Clasificación de imágenes	ResNet, AmoebaNet, EfficientNet, MNasNet, MNIST
			Modelado de lenguajes	Transformer, BERT
			Detección de objetos	RetinaNet
			Segmentación de imágenes	Mask RCNN, ShapeMask
TF 1.15	22 de octubre de 2019	30 de junio de 2020	Clasificación de imágenes	ResNet, AmoebaNet, EfficientNet, MNasNet, MNIST
TF 1.14	19 de julio de 2019	30 de junio de 2020	Clasificación de imágenes	ResNet, AmoebaNet, EfficientNet, MNasNet, MNIST
			Modelado de lenguajes	Transformer
			Detección de objetos	RetinaNet, Mask R-CNN
TF 1.13	11 de marzo de 2019	30 de junio de 2020	Clasificación de imágenes	ResNet, AmoebaNet, MNasNet, MNIST
			Modelado de lenguajes	Transformer
			Detección de objetos	RetinaNet

La versión tf-nightly de TensorFlow solo está disponible para uso experimental. Se compila de noche y puede ser inestable. No es oficialmente compatible y no debe usarse en entornos de producción.

Versión de PyTorch	Inicio de la compatibilidad	Finalización de la compatibilidad	Categoría del modelo	Modelos compatibles
PyTorch 1.x	17 de octubre de 2019	(Fecha de finalización aún no establecida)	Clasificación de imágenes	ResNet
			Modelado de lenguajes	FairSeq Transformer,FairSeq RoBERTa
			Sistemas de recomendación	DLRM