Skip to content

Commit 2884242

Browse files
move distillation runner config to correct file
1 parent e2425f0 commit 2884242

File tree

2 files changed

+21
-15
lines changed

2 files changed

+21
-15
lines changed

source/isaaclab_rl/isaaclab_rl/rsl_rl/distillation_cfg.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,8 @@
1010

1111
from isaaclab.utils import configclass
1212

13+
from .rl_cfg import RslRlBaseRunnerCfg
14+
1315
#########################
1416
# Policy configurations #
1517
#########################
@@ -93,3 +95,22 @@ class RslRlDistillationAlgorithmCfg:
9395

9496
loss_type: Literal["mse", "huber"] = "mse"
9597
"""The loss type to use for the student policy."""
98+
99+
100+
#########################
101+
# Runner configurations #
102+
#########################
103+
104+
105+
@configclass
106+
class RslRlDistillationRunnerCfg(RslRlBaseRunnerCfg):
107+
"""Configuration of the runner for distillation algorithms."""
108+
109+
class_name: str = "DistillationRunner"
110+
"""The runner class name. Default is DistillationRunner."""
111+
112+
policy: RslRlDistillationStudentTeacherCfg = MISSING
113+
"""The policy configuration."""
114+
115+
algorithm: RslRlDistillationAlgorithmCfg = MISSING
116+
"""The algorithm configuration."""

source/isaaclab_rl/isaaclab_rl/rsl_rl/rl_cfg.py

Lines changed: 0 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010

1111
from isaaclab.utils import configclass
1212

13-
from .distillation_cfg import RslRlDistillationAlgorithmCfg, RslRlDistillationStudentTeacherCfg
1413
from .rnd_cfg import RslRlRndCfg
1514
from .symmetry_cfg import RslRlSymmetryCfg
1615

@@ -237,17 +236,3 @@ class RslRlOnPolicyRunnerCfg(RslRlBaseRunnerCfg):
237236

238237
algorithm: RslRlPpoAlgorithmCfg = MISSING
239238
"""The algorithm configuration."""
240-
241-
242-
@configclass
243-
class RslRlDistillationRunnerCfg(RslRlBaseRunnerCfg):
244-
"""Configuration of the runner for distillation algorithms."""
245-
246-
class_name: str = "DistillationRunner"
247-
"""The runner class name. Default is DistillationRunner."""
248-
249-
policy: RslRlDistillationStudentTeacherCfg = MISSING
250-
"""The policy configuration."""
251-
252-
algorithm: RslRlDistillationAlgorithmCfg = MISSING
253-
"""The algorithm configuration."""

0 commit comments

Comments
 (0)