LoadAllTPUEmbeddingParameters

kelas akhir publik LoadAllTPUEmbeddingParameters

Sebuah operasi yang memuat parameter pengoptimalan ke dalam memori penyematan.

Sebuah operasi yang memuat parameter pengoptimalan ke dalam memori penyematan. 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. Untuk Adagrad, auxiliary1 harus menjadi akumulator. Untuk SGD, semua nilai tambahan* harus kosong. Untuk FTRL, auxiliary1 harus berupa akumulator dan auxiliary2 harus berupa suku linier. Untuk ADAM, bantu1 harus menjadi momentum dan bantu2 harus menjadi kecepatan.

Metode Publik

LoadAllTPUEmbeddingParameter statis
buat ( Lingkup lingkup, Iterable< Operand <Float>> parameter, Iterable< Operand <Float>> pembantu1, Iterable< Operand <Float>> pembantu2, Iterable< Operand <Float>> pembantu3, Iterable< Operand <Float>> pembantu4, Dapat diubah< Operan <Float>> tambahan5, Dapat diubah< Operan <Float>> tambahan6, Dapat diubah< Operan <Float>> tambahan7, Konfigurasi string, Jumlah panjang Pecahan, Panjang shardId)
Metode pabrik untuk membuat kelas yang membungkus operasi LoadAllTPUEmbeddingParameters baru.

Metode Warisan

Metode Publik

LoadAllTPUEmbeddingParameters statis publik membuat ( Lingkup lingkup, Iterable< Operand <Float>> parameter, Iterable< Operand <Float>> pembantu1, Iterable< Operand <Float>> pembantu2, Iterable< Operand <Float>> pembantu3, Iterable< Operand <Float> > pembantu4, Iterable< Operan <Float>> pembantu5, Iterable< Operan <Float>> pembantu6, Iterable< Operan <Float>> pembantu7, Konfigurasi string, Jumlah panjang Pecahan, Panjang shardId)

Metode pabrik untuk membuat kelas yang membungkus operasi LoadAllTPUEmbeddingParameters baru.

Parameter
cakupan ruang lingkup saat ini
parameter Daftar tensor, satu untuk setiap tabel penyematan, berisi parameter tabel penyematan awal untuk digunakan dalam pencarian penyematan.
bantu1 Daftar tensor, satu untuk setiap tabel penyematan, berisi nilai awal parameter pengoptimalan tambahan pertama yang digunakan dalam penyematan pembaruan loop pelatihan. Bentuk setiap entri diabaikan (dan karenanya bisa kosong) untuk tabel yang algoritma optimasinya tidak memiliki setidaknya satu parameter tambahan.
tambahan2 Daftar tensor, satu untuk setiap tabel penyematan, berisi nilai awal parameter pengoptimalan tambahan kedua yang akan digunakan dalam penyematan pembaruan loop pelatihan. Bentuk setiap entri diabaikan (dan karenanya bisa kosong) untuk tabel-tabel yang algoritma optimasinya tidak memiliki setidaknya dua tabel tambahan.
tambahan3 Daftar tensor, satu untuk setiap tabel penyematan, berisi nilai awal parameter pengoptimalan tambahan ketiga yang akan digunakan dalam penyematan pembaruan loop pelatihan. Bentuk setiap entri diabaikan (dan karenanya bisa kosong) untuk tabel yang algoritma optimasinya tidak memiliki tiga parameter tambahan.
tambahan4 Daftar tensor, satu untuk setiap tabel penyematan, berisi nilai awal parameter pengoptimalan tambahan kedua yang akan digunakan dalam penyematan pembaruan loop pelatihan. Bentuk setiap entri diabaikan (dan karenanya bisa kosong) untuk tabel-tabel yang algoritma optimasinya tidak memiliki setidaknya empat tabel tambahan.
tambahan5 Daftar tensor, satu untuk setiap tabel penyematan, berisi nilai awal parameter pengoptimalan tambahan ketiga yang akan digunakan dalam penyematan pembaruan loop pelatihan. Bentuk setiap entri diabaikan (dan karenanya bisa kosong) untuk tabel yang algoritma optimasinya tidak memiliki lima parameter tambahan.
tambahan6 Daftar tensor, satu untuk setiap tabel penyematan, berisi nilai awal parameter pengoptimalan tambahan kedua yang akan digunakan dalam penyematan pembaruan loop pelatihan. Bentuk setiap entri diabaikan (dan karenanya bisa kosong) untuk tabel-tabel yang algoritma optimasinya tidak memiliki setidaknya enam tabel tambahan.
tambahan7 Daftar tensor, satu untuk setiap tabel penyematan, berisi nilai awal parameter pengoptimalan tambahan ketiga yang akan digunakan dalam penyematan pembaruan loop pelatihan. Bentuk setiap entri diabaikan (dan karenanya bisa kosong) untuk tabel yang algoritma optimasinya tidak memiliki parameter tambahan sevan.
konfigurasi Proto TPUEmbeddingConfiguration yang menjelaskan parameter tabel yang dimuat, diserialkan ke string.
jumlah pecahan Jumlah pecahan yang menjadi tempat pembagian tabel penyematan.
pecahanId Pengidentifikasi pecahan untuk operasi ini.
Kembali
  • contoh baru LoadAllTPUEmbeddingParameters