forgot to add gamma to custom_objects, relevant for finetuning

andreped · Aug 21, 2022 · ade86e4 · ade86e4
1 parent 534c212
commit ade86e4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/train_agent.py b/src/train_agent.py
@@ -63,6 +63,7 @@ def train_with_masks(ret):
                 "learning_rate": ret.learning_rate,
                 "batch_size": ret.batch_size,
                 "clip_range": lambda _: 0.2,
+                "gamma": ret.gamma,
             }
 
         print("Finetuning...")