Exemple #1
0
def register_custom_components():
    global_env_registry().register_env(
        env_name_prefix='my_custom_env_',
        make_env_func=make_custom_env_func,
        add_extra_params_func=add_extra_params_func,
        override_default_params_func=override_default_params_func,
    )

    register_custom_encoder('custom_env_encoder', CustomEncoder)
Exemple #2
0
def register_models():
    quad_custom_encoder_name = 'quad_multi_encoder'
    if quad_custom_encoder_name not in ENCODER_REGISTRY:
        register_custom_encoder(quad_custom_encoder_name, QuadMultiEncoder)
Exemple #3
0
def register_models():
    register_custom_encoder('vizdoom', VizdoomEncoder)
Exemple #4
0
def dmlab_register_models():
    register_custom_encoder('dmlab_instructions', DmlabEncoder)