tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Merge branch 'main' into user/adil-zouitine/2025-1-7-port-hil-serl-new

Browse Source
This commit is contained in:
Adil Zouitine
2025-05-06 16:43:44 +02:00
committed by AdilZouitine
parent 5998203a33 adcb07bf62
commit e76f29ff7a
272 changed files with 14894 additions and 11422 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/optim/test_schedulers.py
View File

@@ -37,7 +37,6 @@ def test_diffuser_scheduler(optimizer):
"base_lrs": [0.001],
"last_epoch": 1,
"lr_lambdas": [None],
"verbose": False,
}
assert scheduler.state_dict() == expected_state_dict
@@ -56,7 +55,6 @@ def test_vqbet_scheduler(optimizer):
"base_lrs": [0.001],
"last_epoch": 1,
"lr_lambdas": [None],
"verbose": False,
}
assert scheduler.state_dict() == expected_state_dict
@@ -77,7 +75,6 @@ def test_cosine_decay_with_warmup_scheduler(optimizer):
"base_lrs": [0.001],
"last_epoch": 1,
"lr_lambdas": [None],
"verbose": False,
}
assert scheduler.state_dict() == expected_state_dict