Muat parameter penyematan Momentum dengan dukungan debug.
Sebuah operasi yang memuat parameter pengoptimalan ke dalam HBM untuk disematkan. Harus didahului dengan operasi ConfigureTPUEmbeddingHost yang menyiapkan konfigurasi tabel penyematan yang benar. Misalnya, operasi ini digunakan untuk menginstal parameter yang dimuat dari pos pemeriksaan sebelum loop pelatihan dijalankan.
Kelas Bersarang
kelas | LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options | Atribut opsional untuk LoadTPUEmbeddingMomentumParametersGradAccumDebug |
Metode Publik
LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options statis | konfigurasi (Konfigurasi string) |
LoadTPUEmbeddingMomentumParametersGradAccumDebug statis | |
LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options statis | tableId (Id tabel panjang) |
LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options statis | nama meja (String nama tabel) |
Metode Warisan
Metode Publik
Konfigurasi LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options statis publik (Konfigurasi string)
LoadTPUEmbeddingMomentumParametersGradAccumDebug statis publik buat ( Cakupan cakupan , parameter Operand <Float>, momen Operand <Float>, GradientAccumulator Operand <Float>, Long numShards, Long shardId, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi LoadTPUEmbeddingMomentumParametersGradAccumDebug baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
parameter | Nilai parameter yang digunakan dalam algoritma optimasi Momentum. |
momentum | Nilai momentum digunakan pada algoritma optimasi Momentum. |
akumulator gradien | Nilai gradien_akumulator yang digunakan dalam algoritma optimasi Momentum. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru LoadTPUEmbeddingMomentumParametersGradAccumDebug