We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 448af9d commit 8075c48Copy full SHA for 8075c48
src/tests/test_finetuning.py
@@ -246,8 +246,6 @@ def test_finetuning_weight_decay(
246
args, kwargs = train.call_args
247
optimizer = args[4]
248
249
- print(optimizer.state_dict())
250
-
251
assert isinstance(optimizer, AdamW)
252
assert optimizer.state_dict()["param_groups"][0]["weight_decay"] == approx(0.01)
253
src/tests/utils.py
@@ -20,7 +20,6 @@ def __init__(self):
20
self.padding_side = "left"
21
22
def __call__(self, *args, **kwargs):
23
- print(f"{kwargs=}")
24
ids = self.encode(*args, **kwargs)
25
return {"input_ids": ids}
26
0 commit comments