SendTPUEmbeddingGradients

الفئة النهائية العامة SendTPUEmbeddingGradients

إجراء تحديثات متدرجة لجداول التضمين.

الأساليب العامة

ثابت SendTPUEmbeddingGradients
إنشاء (نطاق النطاق ، مدخلات Iterable < Operand <Float>>، Iterable < Operand <Float>> LearningRates، تكوين السلسلة)
طريقة المصنع لإنشاء فئة تغلف عملية SendTPUEmbeddingGradients جديدة.

الطرق الموروثة

الأساليب العامة

إنشاء SendTPUEmbeddingGradients الثابت العام ( نطاق النطاق ، مدخلات Iterable< Operand <Float>>، Iterable< Operand <Float>> LearningRates، تكوين السلسلة)

طريقة المصنع لإنشاء فئة تغلف عملية SendTPUEmbeddingGradients جديدة.

حدود
نِطَاق النطاق الحالي
المدخلات قائمة TensorList للتدرجات التي يمكن من خلالها تحديث جداول التضمين. تحتوي هذه الوسيطة على نفس الطول والأشكال مثل القيمة المرجعة لـ RecvTPUEmbeddingActivations، ولكنها تحتوي على تدرجات لفقد النموذج فيما يتعلق بعمليات تنشيط التضمين. يتم تحديث جداول التضمين من هذه التدرجات عبر المحسن المحدد في تكوين تضمين TPU المعطى لـ tpu.initialize_system.
LearningRates قائمة TensorList من وحدات القياس float32، واحدة لكل علامة معدل تعلم ديناميكي: راجع التعليقات في //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. يمكن لجداول متعددة مشاركة نفس علامة معدل التعلم الديناميكي كما هو محدد في التكوين. إذا كانت معدلات التعلم لجميع الجداول ثابتة، فيجب أن تكون هذه القائمة فارغة.
التكوين نموذج متسلسل لتكوين TPUEmbedding.
عائدات
  • مثيل جديد لـ SendTPUEmbeddingGradients