Skip to content

lit_sb3.modules

[docs] package lit_sb3.modules 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
from lit_sb3.modules.batch_renorm import BatchRenorm, BatchRenorm1d
from lit_sb3.modules.feature_extractors.base import BaseFeaturesExtractor
from lit_sb3.modules.feature_extractors.combined import CombinedExtractor
from lit_sb3.modules.feature_extractors.flatten import FlattenExtractor
from lit_sb3.modules.feature_extractors.nature_cnn import NatureCNN
from lit_sb3.modules.mlp import MlpExtractor, create_mlp, get_actor_critic_arch

__all__ = [
    "BaseFeaturesExtractor",
    "BatchRenorm",
    "BatchRenorm1d",
    "CombinedExtractor",
    "FlattenExtractor",
    "MlpExtractor",
    "NatureCNN",
    "create_mlp",
    "get_actor_critic_arch",
]