Tải các tham số nhúng ADAM.
Một op tải các tham số tối ưu hóa vào HBM để nhúng. Trước đó phải có op configureTPUEmbeddingHost để thiết lập cấu hình bảng nhúng chính xác. Ví dụ: op này được sử dụng để cài đặt các tham số được tải từ điểm kiểm tra trước khi vòng lặp huấn luyện được thực thi.
Các lớp lồng nhau
lớp học | LoadTPUEembeddingADAMParameters.Options | Các thuộc tính tùy chọn cho LoadTPUEmbeddingADAMParameters |
Phương pháp công khai
LoadTPUEembeddingADAMParameters.Options tĩnh | config (Cấu hình chuỗi) |
Tải tĩnhTPUEnhúngADAMThông số | tạo ( Phạm vi phạm vi, tham số Toán hạng <Float>, khoảnh khắc Toán hạng <Float>, Vận tốc toán hạng <Float>, numShards dài, shardId dài, Tùy chọn... tùy chọn) Phương thức ban đầu để tạo một lớp bao gồm một hoạt động LoadTPUEmbeddingADAMParameters mới. |
LoadTPUEembeddingADAMParameters.Options tĩnh | tableId (Id bảng dài) |
LoadTPUEembeddingADAMParameters.Options tĩnh | tên bảng (Tên bảng chuỗi) |
Phương pháp kế thừa
Phương pháp công khai
công khai LoadTPUEmbeddingADAMParameters tạo (Phạm vi phạm vi, Tham số toán hạng <Float>, Momenta toán hạng <Float>, vận tốc toán hạng <Float>, numShards dài, shardId dài, Tùy chọn... tùy chọn)
Phương thức ban đầu để tạo một lớp bao gồm một hoạt động LoadTPUEmbeddingADAMParameters mới.
Thông số
phạm vi | phạm vi hiện tại |
---|---|
thông số | Giá trị của các tham số được sử dụng trong thuật toán tối ưu hóa ADAM. |
khoảnh khắc | Giá trị mô men được sử dụng trong thuật toán tối ưu hóa ADAM. |
vận tốc | Giá trị vận tốc được sử dụng trong thuật toán tối ưu hóa ADAM. |
tùy chọn | mang các giá trị thuộc tính tùy chọn |
Trả lại
- một phiên bản mới của LoadTPUEmbeddingADAMParameters