qihuanye/lewm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Parameter Tuning

Browse Source
This commit is contained in:
qihuanye
2026-05-04 08:20:23 +00:00
parent e84074d6d6
commit f08f2b82f4
2 changed files with 135 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
config/eval/solver/gradient.yaml
View File

@@ -1,8 +1,8 @@
_target_: stable_worldmodel.solver.GradientSolver
model: ???
# Original adam.yaml reference: n_steps=30, num_samples=100, batch_size=1, lr=0.1.
n_steps: 125
batch_size: 50
n_steps: 90
batch_size: 100
num_samples: 1
action_noise: 0
device: "cuda"
@@ -11,4 +11,4 @@ optimizer_cls:
_target_: hydra.utils.get_class
path: torch.optim.AdamW
optimizer_kwargs:
lr: 0.05
lr: 0.075