GönderTPUEkatıştırmaDegradeler

genel final sınıfı SendTPUEmbeddingGradients

Gömme tablolarının degrade güncellemelerini gerçekleştirir.

Genel Yöntemler

statik SendTPUEkatıştırmaDegradeleri
create ( Kapsam kapsamı, Yinelenebilir< İşlenen <Float>> girişler, Yinelenebilir< İşlenen <Float>> öğrenmeRates, String yapılandırması)
Yeni bir SendTPUEmbeddingGradients işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.

Kalıtsal Yöntemler

Genel Yöntemler

public static SendTPUEmbeddingGradients create ( Kapsam kapsamı, Yinelenebilir< İşlenen <Float>> girişler, Yinelenebilir< İşlenen <Float>> öğrenmeRates, String yapılandırması)

Yeni bir SendTPUEmbeddingGradients işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.

Parametreler
kapsam mevcut kapsam
girişler Gömme tablolarının güncelleneceği degradelerin TensorList'i. Bu bağımsız değişken, RecvTPUEmbeddingActivations'ın dönüş değeriyle aynı uzunluğa ve şekillere sahiptir ancak yerleştirme etkinleştirmelerine göre modelin kaybının derecelerini içerir. Yerleştirme tabloları, tpu.initialize_system'e verilen TPU yerleştirme yapılandırmasında belirtilen optimize edici aracılığıyla bu degradelerden güncellenir.
öğrenmeOranları Her dinamik öğrenme oranı etiketi için bir tane olmak üzere float32 skalerlerinden oluşan bir TensorList: //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto içindeki yorumlara bakın. Birden fazla tablo, yapılandırmada belirtildiği gibi aynı dinamik öğrenme oranı etiketini paylaşabilir. Tüm tabloların öğrenme oranları sabitse bu liste boş olmalıdır.
yapılandırma Serileştirilmiş TPUEmbedddingConfiguration protokolü.
İadeler
  • SendTPUEmbeddingGradients'in yeni bir örneği