File tree Expand file tree Collapse file tree 2 files changed +21
-15
lines changed
source/isaaclab_rl/isaaclab_rl/rsl_rl Expand file tree Collapse file tree 2 files changed +21
-15
lines changed Original file line number Diff line number Diff line change 10
10
11
11
from isaaclab .utils import configclass
12
12
13
+ from .rl_cfg import RslRlBaseRunnerCfg
14
+
13
15
#########################
14
16
# Policy configurations #
15
17
#########################
@@ -93,3 +95,22 @@ class RslRlDistillationAlgorithmCfg:
93
95
94
96
loss_type : Literal ["mse" , "huber" ] = "mse"
95
97
"""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."""
Original file line number Diff line number Diff line change 10
10
11
11
from isaaclab .utils import configclass
12
12
13
- from .distillation_cfg import RslRlDistillationAlgorithmCfg , RslRlDistillationStudentTeacherCfg
14
13
from .rnd_cfg import RslRlRndCfg
15
14
from .symmetry_cfg import RslRlSymmetryCfg
16
15
@@ -237,17 +236,3 @@ class RslRlOnPolicyRunnerCfg(RslRlBaseRunnerCfg):
237
236
238
237
algorithm : RslRlPpoAlgorithmCfg = MISSING
239
238
"""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."""
You can’t perform that action at this time.
0 commit comments