Merge branch 'main' of https://github.com/gncs/botnet

Author: ilyes319

scripts/run_train.py

@@ -215,21 +215,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