Fixes unexpected keyword argument learning_rate within RandomNetworkDistillation (#87)

* Removes learning_rate from the dict argument of RandomNetworkDistillation

* Updates contributors

* Uses pop to remove learning_rate rather than using a dict copy

Author: ozhanozen

CONTRIBUTORS.md

@@ -33,6 +33,7 @@ Please keep the lists sorted alphabetically.
 * Lorenzo Terenzi
 * Marko Bjelonic
 * Matthijs van der Boon
+* Özhan Özen
 * Pascal Roth
 * Zhang Chong
 * Ziqi Fan

rsl_rl/algorithms/ppo.py

@@ -59,11 +59,13 @@ def __init__(
 
         # RND components
         if rnd_cfg is not None:
+            # Extract learning rate and remove it from the original dict
+            learning_rate = rnd_cfg.pop("learning_rate", 1e-3)
             # Create RND module
             self.rnd = RandomNetworkDistillation(device=self.device, **rnd_cfg)
             # Create RND optimizer
             params = self.rnd.predictor.parameters()
-            self.rnd_optimizer = optim.Adam(params, lr=rnd_cfg.get("learning_rate", 1e-3))
+            self.rnd_optimizer = optim.Adam(params, lr=learning_rate)
         else:
             self.rnd = None
             self.rnd_optimizer = None