Hacktoberfest 2026: những issue maintainer đã đánh dấu cho tháng Mười, đang mở và phù hợp người mới. Xem issue Hacktoberfest

feat(tf2): align optimizer dispatch with the common schema

Đang mở
  • Có gì không ổn?
#5,878 0 bình luận 0 reaction 0 người được giao Xem trên GitHub

Maintainer thường phản hồi trong vòng 3 ngày

Chưa có ai nhận issue này.

Đánh giá

Độ khó
5/5
Thời gian dự kiến
Hơn một tuần
Mức phù hợp với người mới
35/100
Loại issue
Tính năng
Độ rõ ràng
Khá rõ ràng
Mức độ hoạt động
Ít trao đổi
Công nghệ
python, tensorflow
Lĩnh vực
backend, machine-learning, testing

Hướng nghiên cứu

Bắt đầu từ TF2 trainer và ranh giới dispatch/factory dùng chung, sau đó lần theo cách khối optimizer cấp cao nhất đã được chuẩn hóa đi đến bước xác thực backend. Chạy hoặc thêm các kiểm thử cập nhật một bước có tính quyết định cho các tham số được hỗ trợ, đồng thời xác minh rằng checkpoint/resume tái tạo được lần cập nhật tiếp theo. Hoàn tất khi mọi schema optimizer đều có trạng thái TF2 rõ ràng, các lựa chọn không được hỗ trợ đều thất bại trước khi huấn luyện, và HybridMuon có thể được tích hợp thông qua #5771 mà không phân tích cú pháp trùng lặp.

Do mô hình lập chỉ mục viết ra từ nội dung của issue.

Mô tả

enhancement

Summary

TF2 optimizer dispatch supports Adam and AdamW, while the common schema also exposes LKF, AdaMuon, and HybridMuon. Unsupported choices currently fail only in the TF2 trainer.

DPA4 HybridMuon support is tracked in #5771; this issue covers the broader backend-aware optimizer contract, including explicit support/restriction for the remaining schema variants.

Scope

  • Consume and validate the normalized top-level optimizer block.
  • Implement TF2-native optimizers that are intended to be supported.
  • Reject intentionally unsupported variants during backend-aware validation.
  • Wire optimizer-specific parameters and preserve state through checkpoints.
  • Do not silently fall back to Adam or ignore normalized fields.
  • Coordinate the shared dispatch/factory boundary with #5771.

Acceptance criteria

  • Every schema-exposed optimizer has an explicit TF2 support status.
  • Supported optimizer parameters affect deterministic one-step update tests.
  • Unsupported variants fail before the training loop with an actionable message.
  • Adam and AdamW remain behaviorally compatible.
  • Checkpoint/resume restores complete optimizer state and reproduces the next update.
  • HybridMuon integration can land through #5771 without duplicating configuration parsing.

Refs #5757.

Coding agent: Codex
Codex version: codex-cli 0.144.4
Model: gpt-5.6-sol
Reasoning effort: xhigh

Ngôn ngữ chính
Python
Star
2.1k
Fork
659
Merge trung bình
4 ngày 2 giờ
Pull request đã merge (30 ngày)
6

Chuẩn bị môi trường

Mở trong Codespaces

Khởi chạy dev container của dự án ngay trên trình duyệt, bằng tài khoản GitHub của bạn.

Bắt đầu từ đâu

  1. Đọc hết issue, rồi đọc hướng dẫn đóng góp của dự án.
  2. Bình luận trên issue rằng bạn sẽ nhận — tránh hai người làm cùng một việc.
  3. Fork repository và làm thay đổi trên một nhánh.
  4. Mở pull request có tham chiếu số hiệu của issue.

Issue khác của deepmodeling/deepmd-kit

Tất cả issue của deepmodeling/deepmd-kit

Issue tương tự

Thêm issue về Python

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.