genel final sınıfı SendTPUEmbeddingGradients
Gömme tablolarının degrade güncellemelerini gerçekleştirir.
Genel Yöntemler
statik SendTPUEkatıştırmaDegradeleri |
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