Tải các tham số nhúng Momentum có hỗ trợ gỡ lỗi.
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 | LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options | Các thuộc tính tùy chọn cho LoadTPUEmbeddingMomentumParametersGradAccumDebug |
Phương pháp công khai
LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tĩnh | config (Cấu hình chuỗi) |
LoadTPUEmbeddingMomentumParametersGradAccumDebug | |
LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tĩnh | tableId (Id bảng dài) |
LoadTPUEmbeddingMomentumParametersGradAccumDebug.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 LoadTPUEmbeddingMomentumParametersGradAccumDebug tạo ( Phạm vi phạm vi , Tham số toán hạng <Float>, Khoảnh khắc toán hạng <Float>, Toán hạng <Float> gradientAccumulators, Tùy chọn numShards dài, shardId dài, Tùy chọn... )
Phương thức ban đầu để tạo một lớp bao gồm một thao tác LoadTPUEmbeddingMomentumParametersGradAccumDebug 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 Động lượng. |
khoảnh khắc | Giá trị của xung lượng được sử dụng trong thuật toán tối ưu hóa xung lượng. |
gradientTích lũy | Giá trị của gradient_accumulators được sử dụng trong thuật toán tối ưu hóa Động lượng. |
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 LoadTPUEmbeddingMomentumParametersGradAccumDebug