Update comment for AdamW implementation reference. (#4604)

Co-authored-by: Will Findley <findley@gmail.com>
This commit is contained in:
WillCorticesAI 2023-12-26 05:42:08 -05:00 committed by GitHub
parent 77465dad48
commit de8e496437
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

4
ggml.c
View File

@ -17456,9 +17456,9 @@ static void ggml_opt_acc_grad(int np, struct ggml_tensor * const ps[], float * g
}
//
// ADAM
// Using AdamW - ref: https://arxiv.org/pdf/1711.05101v3.pdf
//
// ref: https://arxiv.org/pdf/1412.6980.pdf
// (Original Adam - ref: https://arxiv.org/pdf/1412.6980.pdf)
//
static enum ggml_opt_result ggml_opt_adam(