simplify parameter dict for optimizers

Author: gncs

scripts/run_train.py

@@ -213,21 +213,10 @@ def main() -> None:
 
     # Optimizer
     param_options = dict(
-        params=[{
-            'name': 'embedding',
-            'params': model.node_embedding.parameters(),
-            'weight_decay': 0.0,
-        }, {
-            'name': 'interactions',
-            'params': model.interactions.parameters(),
-            'weight_decay': args.weight_decay,
-        }, {
-            'name': 'readouts',
-            'params': model.readouts.parameters(),
-            'weight_decay': 0.0,
-        }],
+        params=model.parameters(),
         lr=args.lr,
         amsgrad=args.amsgrad,
+        weight_decay=args.weight_decay,
     )
 
     optimizer: torch.optim.Optimizer