Compare commits
7 Commits
user/miche
...
user/azoui
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
0961836de5 | ||
|
|
dcb24837e1 | ||
|
|
a75d00970f | ||
|
|
c7b204f4a6 | ||
|
|
4df18de636 | ||
|
|
8dc69c6126 | ||
|
|
7d481e6048 |
@@ -49,7 +49,7 @@ dependencies = [
|
|||||||
"datasets>=2.19.0",
|
"datasets>=2.19.0",
|
||||||
"deepdiff>=7.0.1",
|
"deepdiff>=7.0.1",
|
||||||
"diffusers>=0.27.2",
|
"diffusers>=0.27.2",
|
||||||
"draccus>=0.10.0",
|
"draccus==0.10.0",
|
||||||
"einops>=0.8.0",
|
"einops>=0.8.0",
|
||||||
"flask>=3.0.3",
|
"flask>=3.0.3",
|
||||||
"gdown>=5.1.0",
|
"gdown>=5.1.0",
|
||||||
|
|||||||
@@ -37,7 +37,6 @@ def test_diffuser_scheduler(optimizer):
|
|||||||
"base_lrs": [0.001],
|
"base_lrs": [0.001],
|
||||||
"last_epoch": 1,
|
"last_epoch": 1,
|
||||||
"lr_lambdas": [None],
|
"lr_lambdas": [None],
|
||||||
"verbose": False,
|
|
||||||
}
|
}
|
||||||
assert scheduler.state_dict() == expected_state_dict
|
assert scheduler.state_dict() == expected_state_dict
|
||||||
|
|
||||||
@@ -56,7 +55,6 @@ def test_vqbet_scheduler(optimizer):
|
|||||||
"base_lrs": [0.001],
|
"base_lrs": [0.001],
|
||||||
"last_epoch": 1,
|
"last_epoch": 1,
|
||||||
"lr_lambdas": [None],
|
"lr_lambdas": [None],
|
||||||
"verbose": False,
|
|
||||||
}
|
}
|
||||||
assert scheduler.state_dict() == expected_state_dict
|
assert scheduler.state_dict() == expected_state_dict
|
||||||
|
|
||||||
@@ -77,7 +75,6 @@ def test_cosine_decay_with_warmup_scheduler(optimizer):
|
|||||||
"base_lrs": [0.001],
|
"base_lrs": [0.001],
|
||||||
"last_epoch": 1,
|
"last_epoch": 1,
|
||||||
"lr_lambdas": [None],
|
"lr_lambdas": [None],
|
||||||
"verbose": False,
|
|
||||||
}
|
}
|
||||||
assert scheduler.state_dict() == expected_state_dict
|
assert scheduler.state_dict() == expected_state_dict
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user