MCPcopy Create free account

hub / github.com/SurrealAI/surreal / functions

Functions791 in github.com/SurrealAI/surreal

↓ 40 callersMethodreset
Resets the state of the environment and returns an initial observation. Returns: observation (object): the initial observation of the
surreal/env/base.py:96
↓ 40 callersMethodstep
Run one timestep of the environment's dynamics. When end of episode is reached, you are responsible for calling `reset()` to reset thi
surreal/env/base.py:78
↓ 40 callersMethodtime
(self)
surreal/utils/common.py:526
↓ 29 callersMethodjoin
(self)
surreal/replay/base.py:62
↓ 24 callersMethodextend
(self, default_config)
surreal/session/config.py:229
↓ 24 callersMethodget_command
(self, role)
surreal/launch/command_generator.py:30
↓ 23 callersFunctionextend_config
default_config must specify all the expected keys. Use the following special values for required placeholders: * _req_: require a single
surreal/session/config.py:237
↓ 18 callersMethodsum
Returns arr[start] + ... + arr[end]
surreal/replay/segment_tree.py:97
↓ 15 callersMethodstart
(self)
surreal/utils/common.py:534
↓ 14 callersFunctionpytest_print_raises
(exc)
test-old/utils.py:9
↓ 12 callersMethodstep
(self, action)
surreal/env/wrapper.py:52
↓ 10 callersMethodnumpy
(self, dtype=None)
surreal/env/atari_wrappers.py:210
↓ 9 callersMethodadd_value
(self, value)
surreal/utils/common.py:475
↓ 9 callersMethodget
Returns a SharedMemoryObject whose .data attribute contains data
surreal/distributed/data_fetcher.py:63
↓ 9 callersMethodload
Load data from a numpy_dictionary Args: numpy_dict: { <module_dict_key>: { <para
surreal/distributed/module_dict.py:47
↓ 9 callersMethodmain
Default Main loop Args: @env: the environment to run agent on
surreal/agent/base.py:224
↓ 9 callersMethodrender
Renders the environment.
surreal/env/base.py:105
↓ 8 callersMethod__init__
(self, env)
surreal/env/wrapper.py:24
↓ 8 callersMethodadd_scalars
(self, new_data)
surreal/utils/common.py:637
↓ 8 callersMethodbuild
Build all images and push them
surreal/launch/build_images.py:70
↓ 8 callersMethodparse_args
(self)
surreal/subproc/surreal_subproc.py:59
↓ 8 callersMethodtrack_increment
(self)
surreal/utils/common.py:596
↓ 8 callersMethodupdate
(self, other)
surreal/session/config.py:182
↓ 7 callersMethod__init__
This object ensures that common frames between the observations are only stored once. It exists purely to optimize memory usage which can be h
surreal/env/atari_wrappers_deprecated.py:157
↓ 7 callersMethodclose
Override _close in your subclass to perform any necessary cleanup. Environments will automatically close() themselves when garbage co
surreal/env/base.py:110
↓ 7 callersMethodobservation_spec
(self)
surreal/env/wrapper.py:86
↓ 6 callersMethodckpt_name
(self, suffix)
surreal/utils/checkpoint.py:221
↓ 6 callersMethodforward
(self, x)
surreal/model/q_net.py:112
↓ 6 callersMethodlikelihood
Method computes likelihood of action (a) given probability (prob)
surreal/model/ppo_net.py:42
↓ 6 callersMethodmetadata_path
(self)
surreal/utils/checkpoint.py:218
↓ 5 callersMethod_get_path
(self, fname)
surreal/utils/checkpoint.py:212
↓ 5 callersFunction_trace_key
(dict_trace, key)
surreal/session/config.py:11
↓ 5 callersMethodaction_spec
(self)
surreal/env/wrapper.py:83
↓ 5 callersMethodforward_actor
forward pass actor to generate policy with option to use z-filter Args: obs: dictionary of batched observatio
surreal/model/ppo_net.py:253
↓ 5 callersMethodforward_critic
forward pass critic to generate policy with option to use z-filter Note: assumes input has either shape length 2 or 4 without
surreal/model/ppo_net.py:284
↓ 5 callersFunctionget_tensorplex_client
(client_id, session_config)
surreal/session/tracker.py:193
↓ 5 callersMethodsend
(self, data)
surreal/env/exp_sender_wrapper.py:56
↓ 4 callersMethod__init__
(self, D_in, D_act, hidden_sizes=[300, 200], use_layernorm=True)
surreal/model/model_builders/builders.py:36
↓ 4 callersMethod__init__
Stack k last frames. Returns lazy array, which is much more memory efficient. See Also -------- baselines.common.ata
surreal/env/atari_wrappers.py:149
↓ 4 callersMethod__init__
Default sender configs are in BASE_SESSION_CONFIG['sender'] They contain communication level information Algorithm s
surreal/env/exp_sender_wrapper.py:11
↓ 4 callersMethod_gather_low_dim_input
Concatenate (along 2nd dimension) all the low-dimensional (propioceptive) features from input observation tuple A
surreal/model/ppo_net.py:168
↓ 4 callersFunction_is_req
(value)
surreal/session/config.py:56
↓ 4 callersMethod_load_metadata
(self)
surreal/utils/checkpoint.py:97
↓ 4 callersMethodcopy
(self)
surreal/session/config.py:233
↓ 4 callersMethodcur_value
Returns current moving average, 0 if no records
surreal/utils/common.py:484
↓ 4 callersMethodget_value
(self)
test-old/test_redis.py:15
↓ 4 callersMethodinsert
Add a new experience to the replay. Includes passive evict logic if memory capacity is exceeded. Args: exp_dict:
surreal/replay/base.py:70
↓ 4 callersFunctionintegration_test
(temp_path, config_path, launcher, env='gym:Hal
surreal/test_helpers/integration_test.py:42
↓ 4 callersMethodlaunch
Launches the specific component Args: component_name(str): the process to launch
surreal/launch/launcher.py:58
↓ 4 callersFunctionmake_env
Makes an environment and populates related fields in env_config return env, env_config Args: mode(str): allow differnt kwargs to
surreal/env/make_env.py:41
↓ 4 callersFunctionmake_env_config
Issue: Surreal-tmux gives segfault on running robosuite environments. Cause: On launching a surreal process, an instance of RL environment is
surreal/env/make_env.py:16
↓ 4 callersMethodrun_component
(self, component_name)
surreal/launch/launcher.py:175
↓ 4 callersFunctionsetup_network
Sets up the communication between surreal components using symphony Args: agents, evals (list): list of symphony
surreal/launch/setup_network.py:4
↓ 4 callersMethodto_dict
Recursively convert back to dictionary
surreal/session/config.py:201
↓ 3 callersMethod__init__
The following options are pre-configured --verbosity, or -vvv (number of v's indicate the level of verbosity) --debug: turn o
surreal/utils/common.py:368
↓ 3 callersMethod__init__
(self, env)
surreal/env/monitor.py:20
↓ 3 callersMethod_add_modality
(self, obs)
surreal/env/dm_wrapper.py:47
↓ 3 callersMethod_add_modality
(self, obs)
surreal/env/wrapper.py:172
↓ 3 callersMethod_add_modality
(self, obs, verbose=False)
surreal/env/wrapper.py:225
↓ 3 callersMethod_check_nfs_retrieve_settings
(self)
surreal/kube/surreal_kube.py:546
↓ 3 callersMethod_filtered_obs
(self, obs, verbose=False)
surreal/env/wrapper.py:483
↓ 3 callersMethod_gcloud_download
(self, remote_path, local_path)
surreal/kube/surreal_kube.py:533
↓ 3 callersMethod_scale_image
Given uint8 input from the environment, scale to float32 and divide by 255 to scale inputs between 0.0 and 1.0 Ar
surreal/model/ppo_net.py:368
↓ 3 callersMethod_step
(self, action)
surreal/env/monitor.py:35
↓ 3 callersMethodadd_value
(self, x)
test-old/test_redis.py:11
↓ 3 callersFunctionassert_type
(x, expected_type, message='')
surreal/utils/common.py:37
↓ 3 callersFunctionbuild_from_config
(config)
surreal/env/mujocomanip/object_builder.py:3
↓ 3 callersMethodckpt_path
(self, suffix)
surreal/utils/checkpoint.py:228
↓ 3 callersMethoddump_file
(self, file_path)
surreal/session/config.py:216
↓ 3 callersMethodfetch_parameter
Extends base class fetch_parameters to add some logging
surreal/agent/base.py:355
↓ 3 callersMethodforward_perception
(self, obs)
surreal/model/ddpg_net.py:69
↓ 3 callersMethodget_actor_params
Method that returns generator that contains all the parameters from actor, cnn_stem (optional), and rnn_stem (optional)
surreal/model/ppo_net.py:202
↓ 3 callersFunctionget_loggerplex_client
Args: name: file name for the remote log file session_config: see session_config.loggerplex
surreal/session/tracker.py:173
↓ 3 callersMethodjoin
Wait for all parameter server workers to exit (Currently this means they crashed) Note that proxy is a daemon
surreal/distributed/parameter_server.py:101
↓ 3 callersMethodjoin
(self)
surreal/replay/sharded_replay.py:105
↓ 3 callersMethodkl
Method computes KL Divergence of between two probability distributions Note: this is D_KL(prob0 || prob1), not D_KL(prob1 ||
surreal/model/ppo_net.py:48
↓ 3 callersMethodmain_loop
One loop of agent, runs one episode of the environment
surreal/agent/base.py:244
↓ 3 callersMethodmain_setup
Setup before constant looping
surreal/agent/base.py:234
↓ 3 callersMethodmin
Returns min(arr[start], ..., arr[end])
surreal/replay/segment_tree.py:135
↓ 3 callersMethodprocess
(self, frame)
surreal/env/atari_wrappers_deprecated.py:132
↓ 3 callersMethodstep_per_sec
Speedometer, step per second Args: average_episodes: average over the past N episodes
surreal/env/monitor.py:59
↓ 3 callersMethodtrain
(self)
test-old/test_checkpoint.py:72
↓ 2 callersMethod__init__
(self, config)
surreal/env/mujocomanip/mujocomanip_envs.py:6
↓ 2 callersMethod__init__
Build a Segment Tree data structure. https://en.wikipedia.org/wiki/Segment_tree Can be used as regular array, but with two imp
surreal/replay/segment_tree.py:7
↓ 2 callersMethod_batch_obs
Helper function that batches a list of observation (nested dictionary) into one whole dictionary of batched observation
surreal/learner/aggregator.py:186
↓ 2 callersMethod_change_folder
Temporary change save folder, useful for loading from other folders
surreal/utils/checkpoint.py:127
↓ 2 callersMethod_check_version
(self)
surreal/utils/checkpoint.py:81
↓ 2 callersFunction_fill_default_config
(config, default_config, dict_trace)
surreal/session/config.py:100
↓ 2 callersMethod_flatten_obs
(self, obs)
surreal/env/wrapper.py:299
↓ 2 callersMethod_get_ob
(self)
surreal/env/atari_wrappers_deprecated.py:203
↓ 2 callersMethod_get_ob
(self)
surreal/env/atari_wrappers.py:182
↓ 2 callersFunction_get_qualified_type_name
(type_)
surreal/utils/common.py:27
↓ 2 callersMethod_get_tensorplex
Get the periodic tensorplex object Args: @name: The name of the collection of metrics
surreal/agent/base.py:108
↓ 2 callersMethod_grayscale
(self, obs)
surreal/env/wrapper.py:370
↓ 2 callersFunction_merge_setting_dictionaries
(customize, base)
surreal/kube/surreal_kube.py:20
↓ 2 callersMethod_observation
(self, observation)
surreal/env/wrapper.py:109
↓ 2 callersFunction_raise_req_error
(key, value, dict_trace, prefix_msg='')
surreal/session/config.py:70
↓ 2 callersFunction_req_type_check
(value)
surreal/session/config.py:24
↓ 2 callersMethod_reset
(self)
surreal/env/mujocomanip/mujocomanip_envs.py:10
↓ 2 callersMethod_restore
helper to be used in self.restore() and self.restore_full_name() Returns: - ckpt file name if successfully restored
surreal/utils/checkpoint.py:102
next →1–100 of 791, ranked by callers