LoadAllTPUEmbeddingParameters

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

عملية تقوم بتحميل معلمات التحسين في الذاكرة المضمنة.

عملية تقوم بتحميل معلمات التحسين في الذاكرة المضمنة. يجب أن يسبقه عملية ConfigureTPUEmbeddingHost التي تقوم بإعداد التكوين الصحيح لجدول التضمين. على سبيل المثال، يتم استخدام هذه العملية لتثبيت المعلمات التي يتم تحميلها من نقطة التفتيش قبل تنفيذ حلقة التدريب. بالنسبة لـ Adagrad، يجب أن يكون المساعد 1 هو المراكم. بالنسبة لـ SGD، يجب أن تكون كافة القيم المساعدة* فارغة. بالنسبة إلى FTRL، يجب أن تكون auxiliary1 هي المراكم وauxiliary2 يجب أن تكون المصطلحات الخطية. بالنسبة لـ ADAM، يجب أن تكون المساعدة 1 هي العزم والسرعة المساعدة 2.

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

ثابت LoadAllTPUEmbeddingParameters
إنشاء ( نطاق النطاق ، Iterable < المعامل <Float>> المعلمات، Iterable < المعامل <Float>> المساعد 1، Iterable < المعامل <Float>> المساعد 2، Iterable < المعامل <Float>> المساعد 3، Iterable < المعامل <Float>> المساعد 4، Iterable< Operand <Float>> auxiliary5، Iterable< Operand <Float>> auxiliary6، Iterable< Operand <Float>> auxiliary7، سلسلة التكوين، Long numShards، Long shardId)
طريقة المصنع لإنشاء فئة تلتف حول عملية LoadAllTPUEmbeddingParameters الجديدة.

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

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

إنشاء معلمات LoadAllTPUEmbeddingParameters ثابتة عامة ( نطاق النطاق، معلمات Iterable< Operand <Float>>، Iterable< Operand <Float>> auxiliary1، Iterable< Operand <Float>> auxiliary2، Iterable< Operand <Float>> auxiliary3، Iterable< Operand <Float> > auxiliary4، Iterable< Operand <Float>> auxiliary5، Iterable< Operand <Float>> auxiliary6، Iterable< Operand <Float>> auxiliary7، تكوين السلسلة، Long numShards، Long shardId)

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

حدود
نِطَاق النطاق الحالي
حدود قائمة الموترات، واحدة لكل جدول تضمين، تحتوي على معلمات جدول التضمين الأولية لاستخدامها في عمليات البحث عن التضمين.
مساعد1 قائمة الموترات، واحدة لكل جدول تضمين، تحتوي على القيم الأولية لمعلمة التحسين المساعدة الأولى لاستخدامها في تضمين تحديثات حلقة التدريب. يتم تجاهل شكل كل إدخال (وبالتالي يمكن أن يكون فارغًا) لتلك الجداول التي لا تحتوي خوارزميات التحسين الخاصة بها على معلمة مساعدة واحدة على الأقل.
مساعد2 قائمة الموترات، واحدة لكل جدول تضمين، تحتوي على القيم الأولية لمعلمة التحسين المساعدة الثانية لاستخدامها في تضمين تحديثات حلقة التدريب. يتم تجاهل شكل كل إدخال (وبالتالي يمكن أن يكون فارغًا) لتلك الجداول التي لا تحتوي خوارزميات التحسين الخاصة بها على عنصرين مساعدين على الأقل
مساعد3 قائمة الموترات، واحدة لكل جدول تضمين، تحتوي على القيم الأولية لمعلمة التحسين المساعدة الثالثة لاستخدامها في تضمين تحديثات حلقة التدريب. يتم تجاهل شكل كل إدخال (وبالتالي يمكن أن يكون فارغًا) لتلك الجداول التي لا تحتوي خوارزميات التحسين الخاصة بها على ثلاث معلمات مساعدة.
مساعد4 قائمة الموترات، واحدة لكل جدول تضمين، تحتوي على القيم الأولية لمعلمة التحسين المساعدة الثانية لاستخدامها في تضمين تحديثات حلقة التدريب. يتم تجاهل شكل كل إدخال (وبالتالي يمكن أن يكون فارغًا) لتلك الجداول التي لا تحتوي خوارزميات التحسين الخاصة بها على أربعة عناصر مساعدة على الأقل
مساعد5 قائمة الموترات، واحدة لكل جدول تضمين، تحتوي على القيم الأولية لمعلمة التحسين المساعدة الثالثة لاستخدامها في تضمين تحديثات حلقة التدريب. يتم تجاهل شكل كل إدخال (وبالتالي يمكن أن يكون فارغًا) لتلك الجداول التي لا تحتوي خوارزميات التحسين الخاصة بها على خمس معلمات مساعدة.
مساعد6 قائمة الموترات، واحدة لكل جدول تضمين، تحتوي على القيم الأولية لمعلمة التحسين المساعدة الثانية لاستخدامها في تضمين تحديثات حلقة التدريب. يتم تجاهل شكل كل إدخال (وبالتالي يمكن أن يكون فارغًا) لتلك الجداول التي لا تحتوي خوارزميات التحسين الخاصة بها على ستة عناصر مساعدة على الأقل
مساعد7 قائمة الموترات، واحدة لكل جدول تضمين، تحتوي على القيم الأولية لمعلمة التحسين المساعدة الثالثة لاستخدامها في تضمين تحديثات حلقة التدريب. يتم تجاهل شكل كل إدخال (وبالتالي يمكن أن يكون فارغًا) لتلك الجداول التي لا تحتوي خوارزميات التحسين الخاصة بها على معلمات مساعدة sevan.
التكوين نموذج أولي لـ TPUEmbeddingConfiguration يصف معلمات الجدول التي يتم تحميلها، ويتم تسلسلها إلى سلسلة.
numShards عدد القطع التي تم تقسيم جداول التضمين إليها.
shardId معرف القطعة لهذه العملية.
عائدات
  • مثيل جديد لـ LoadAllTPUEmbeddingParameters