Muat parameter penyematan SGD.
Operasi yang memuat parameter pengoptimalan ke HBM untuk penyematan. Harus didahului oleh 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 | LoadTPUEbeddingStochasticGradientDescentParameters.Options | Atribut opsional untuk LoadTPUEmbeddingStochasticGradientDescentParameters |
Metode Publik
statis LoadTPUEmbeddingStochasticGradientDescentParameters.Options | config (String config) |
statis LoadTPUEmbeddingStochasticGradientDescentParameters | membuat ( Lingkup lingkup, Operan <Lampung> parameter, Long numShards, Long shardId, Options ... pilihan) Metode pabrik untuk membuat kelas yang membungkus operasi LoadTPUEmbeddingStochasticGradientDescentParameters baru. |
statis LoadTPUEmbeddingStochasticGradientDescentParameters.Options | TableID (Long TableID) |
statis LoadTPUEmbeddingStochasticGradientDescentParameters.Options | tableName (String tableName) |
Metode yang Diwarisi
Metode Publik
public static LoadTPUEmbeddingStochasticGradientDescentParameters membuat ( Scope lingkup, Operan <Lampung> parameter, Long numShards, Long shardId, Options ... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi LoadTPUEmbeddingStochasticGradientDescentParameters baru.
Parameter
cakupan | lingkup saat ini |
---|---|
parameter | Nilai parameter yang digunakan dalam algoritma optimasi penurunan gradien stokastik. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari LoadTPUEmbeddingStochasticGradientDescentParameters