Skip to content

Commit ece2ae1

Browse files
committed
Add multivariate Gaussian model to runner in docs
1 parent bbe532d commit ece2ae1

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

skrl/utils/runner/torch/runner.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,13 @@
1414
from skrl.resources.schedulers.torch import KLAdaptiveLR # noqa
1515
from skrl.trainers.torch import SequentialTrainer, Trainer
1616
from skrl.utils import set_seed
17-
from skrl.utils.model_instantiators.torch import categorical_model, deterministic_model, gaussian_model, shared_model
17+
from skrl.utils.model_instantiators.torch import (
18+
categorical_model,
19+
deterministic_model,
20+
gaussian_model,
21+
multivariate_gaussian_model,
22+
shared_model,
23+
)
1824

1925

2026
class Runner:
@@ -37,6 +43,7 @@ def __init__(self, env: Union[Wrapper, MultiAgentEnvWrapper], cfg: Mapping[str,
3743
"gaussianmixin": gaussian_model,
3844
"categoricalmixin": categorical_model,
3945
"deterministicmixin": deterministic_model,
46+
"multivariategaussianmixin": multivariate_gaussian_model,
4047
"shared": shared_model,
4148
# memory
4249
"randommemory": RandomMemory,

0 commit comments

Comments
 (0)