Don't cherry-pick args

iterative · Aug 7, 2023 · fbf8865 · fbf8865
1 parent 8cac907
commit fbf8865
Showing 1 changed file with 1 addition and 9 deletions.
diff --git a/src/dvclive/huggingface.py b/src/dvclive/huggingface.py
@@ -32,15 +32,7 @@ def on_train_begin(
         control: TrainerControl,
         **kwargs,
     ):
-        for key, value in args.to_dict().items():
-            if key in (
-                "num_train_epochs",
-                "weight_decay",
-                "max_grad_norm",
-                "warmup_ratio",
-                "warmup_steps",
-            ):
-                self.live.log_param(key, value)
+        self.live.log_params(args.to_dict())
 
     def on_log(
         self,