From fbf8865f93d7a906d354d6797b8092d781037f18 Mon Sep 17 00:00:00 2001
From: daavoo <daviddelaiglesiacastro@gmail.com>
Date: Mon, 7 Aug 2023 10:17:21 +0200
Subject: [PATCH] Don't cherry-pick args

---
 src/dvclive/huggingface.py | 10 +---------
 1 file changed, 1 insertion(+), 9 deletions(-)

diff --git a/src/dvclive/huggingface.py b/src/dvclive/huggingface.py
index 21df876e..8847f888 100644
--- 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,