fix learning rate

ASUS-AICS · Mar 5, 2024 · 8d11756 · 8d11756
1 parent 0621e77
commit 8d11756
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/example_config/AmazonCat-13K/attentionxml.yml b/example_config/AmazonCat-13K/attentionxml.yml
@@ -30,8 +30,7 @@ seed: 1337
 epochs: 10
 # https://github.com/Lightning-AI/lightning/issues/8826
 optimizer: adam
-optimizer_config:
-  lr: 0.001
+learning_rate: 0.001
 # early stopping
 patience: 5
 

diff --git a/example_config/EUR-Lex/attentionxml.yml b/example_config/EUR-Lex/attentionxml.yml
@@ -30,6 +30,7 @@ seed: 1337
 epochs: 30
 # https://github.com/Lightning-AI/lightning/issues/8826
 optimizer: adam
+learning_rate: 0.001
 # early stopping
 patience: 5
 

diff --git a/example_config/Wiki10-31K/attentionxml.yml b/example_config/Wiki10-31K/attentionxml.yml
@@ -30,6 +30,7 @@ seed: 1337
 epochs: 30
 # https://github.com/Lightning-AI/lightning/issues/8826
 optimizer: adam
+learning_rate: 0.001
 # early stopping
 patience: 5