[fix]: Correct weight decay parameter logic #10

Closed gaozhun closed 2 months ago