MCPcopy Create free account

hub / github.com/SurrealAI/surreal / functions

Functions791 in github.com/SurrealAI/surreal

↓ 2 callersMethod_stacked_observation
Assumes self._history contains the last n frames from the environment Concatenates the frames together along the depth axis
surreal/env/wrapper.py:414
↓ 2 callersMethod_transpose
(self, obs)
surreal/env/wrapper.py:340
↓ 2 callersMethod_update_endpoint
(self)
surreal/session/tracker.py:22
↓ 2 callersMethodaction_spec
(self)
surreal/env/wrapper.py:471
↓ 2 callersMethodadd
(self, *args, **kwargs)
surreal/utils/common.py:381
↓ 2 callersMethodadd_scalars
Args: tag_value_dict: global_step: None to use the internal counter Returns: - None if period i
surreal/session/tracker.py:133
↓ 2 callersMethodaggregate
Args: exp_list: Returns: aggregated experience
surreal/learner/aggregator.py:52
↓ 2 callersFunctionbinary_hash
Low collision hash of any binary string For designating the 16-char object key in Redis. Runs at 200 mu-second per hash on Macbook pro.
surreal/utils/serializer.py:55
↓ 2 callersFunctionbuild_ffqfunc
(learner_config, env_config)
surreal/model/q_net.py:122
↓ 2 callersMethodclass_name
(cls)
surreal/env/wrapper.py:36
↓ 2 callersMethodclear_actor_grad
Method that clears all gradients from all the parameters from actor, cnn_stem (optional), and rnn_stem (optional)
surreal/model/ppo_net.py:180
↓ 2 callersMethodentropy
Method computes entropy of a given probability (prob)
surreal/model/ppo_net.py:64
↓ 2 callersMethodget_critic_parameters
(self)
surreal/model/ddpg_net.py:57
↓ 2 callersMethodget_critic_params
Method that returns generator that contains all the parameters from critic, cnn_stem (optional), and rnn_stem (optional)
surreal/model/ppo_net.py:214
↓ 2 callersFunctionis_np_array
(L, dtype=None)
surreal/utils/numpy_util.py:57
↓ 2 callersFunctionis_sequence
Returns: True if the sequence is a collections.Sequence and not a string.
surreal/utils/common.py:132
↓ 2 callersMethodload_file
(cls, file_path)
surreal/session/config.py:193
↓ 2 callersMethodobservation_spec
(self)
surreal/env/wrapper.py:459
↓ 2 callersMethodon_fetch_parameter_failed
Called when connection with parameter server fails to be established
surreal/distributed/parameter_server.py:287
↓ 2 callersMethodon_fetch_parameter_success
Called when connection with parameter server is succesfully established
surreal/distributed/parameter_server.py:296
↓ 2 callersMethodperiodic_checkpoint
Will only save at the end of each period Args: global_steps: the number of iterations score: the evaluation
surreal/learner/base.py:277
↓ 2 callersMethodpublish
Called by learner. Publishes model parameters with additional info Args: iteration: current learning iteration
surreal/distributed/parameter_server.py:40
↓ 2 callersMethodpublish_parameter
Learner publishes latest parameters to the parameter server. Args: iteration: the current number of learning iterations
surreal/learner/base.py:136
↓ 2 callersMethodreduce
Returns result of applying `self.operation` to a contiguous subsequence of the array. self.operation(arr[start], operation(arr[sta
surreal/replay/segment_tree.py:50
↓ 2 callersMethodreset
reset of LSTM hidden and cell states
surreal/agent/ppo_agent.py:169
↓ 2 callersMethodrestore
Args: target: can be one of the following semantics - int: 0 for the last (or best), 1 for the second last (or best
surreal/utils/checkpoint.py:141
↓ 2 callersMethodsample
This function is called in _sample_handler for learner side Zmq request Args: batch_size: Returns:
surreal/replay/base.py:80
↓ 2 callersMethodsave
Saves checkpoint to disk Args: file_path: locatioin to save
surreal/learner/base.py:66
↓ 2 callersMethodsetup_agent
Same as launch_agent, but instead of running agent.main() infinite loop, returns the agent instance Args: agent
surreal/launch/launcher.py:192
↓ 2 callersMethodsetup_learner
Same as run_learner, but returns the Learner instance instead of calling learner.main_loop()
surreal/launch/launcher.py:317
↓ 2 callersMethodstart_threads
(self)
surreal/replay/base.py:50
↓ 2 callersMethodtrack_increment
Returns: True if we enter the next period
surreal/session/tracker.py:30
↓ 2 callersMethodupdate_target_params
updates kept parameters to that of another model Args: net: another PPO_Model instance
surreal/model/ppo_net.py:226
↓ 1 callersMethod__call__
(cls, *args, **kwargs)
surreal/utils/common.py:216
↓ 1 callersMethod__init__
Args: folder: checkpoint folder path name: prefix of the checkpoint tracked_obj: arbitrary object whose c
surreal/utils/checkpoint.py:35
↓ 1 callersMethod__init__
(self, env, is_pixel_input)
surreal/env/dm_wrapper.py:37
↓ 1 callersMethod__setitem__
(self, idx, val)
surreal/replay/segment_tree.py:72
↓ 1 callersMethod_action
(self, action)
surreal/env/wrapper.py:133
↓ 1 callersMethod_adapt_loss
Computes the loss with current data. also returns a dictionary of statistics which includes surrogate loss, clipped surrogate los, po
surreal/learner/ppo.py:250
↓ 1 callersMethod_adapt_update
Method that makes policy updates. calls _adapt_loss method Note: self.clip_actor_gradient determines whether gradient is clipped
surreal/learner/ppo.py:287
↓ 1 callersMethod_add_dry_run
(self, parser)
surreal/tmux/surreal_tmux.py:96
↓ 1 callersMethod_add_dry_run
(self, parser)
surreal/kube/surreal_kube.py:169
↓ 1 callersMethod_assert_conforms_to_spec
(self, obs)
surreal/env/wrapper.py:70
↓ 1 callersFunction_camelcase_names
(names)
surreal/env/atari_names.py:65
↓ 1 callersMethod_check_tracked_attrs
(self, tracked_attrs)
surreal/utils/checkpoint.py:89
↓ 1 callersMethod_clip_loss
Computes the loss with current data. also returns a dictionary of statistics which includes surrogate loss, clipped surrogate los, po
surreal/learner/ppo.py:194
↓ 1 callersMethod_clip_update
Method that makes policy updates. calls _clip_loss method Note: self.clip_actor_gradient determines whether gradient is clipped
surreal/learner/ppo.py:227
↓ 1 callersMethod_close
(self)
surreal/env/base.py:65
↓ 1 callersMethod_copy_last_to_best
(self, suffix)
surreal/utils/checkpoint.py:224
↓ 1 callersMethod_ensure_no_double_wrap
(self)
surreal/env/wrapper.py:39
↓ 1 callersMethod_find_executable
Finds the .py file corresponding to the algorithm specified Args: name: ddpg / ppo / <path in container to compatibl
surreal/subproc/surreal_subproc.py:189
↓ 1 callersMethod_find_executable
Finds the .py file corresponding to the algorithm specified Args: name: ddpg / ppo / <path in container to compatibl
surreal/tmux/surreal_tmux.py:196
↓ 1 callersMethod_find_executable
Finds the .py file corresponding to the algorithm specified Args: name: ddpg / ppo / <path in container to compatibl
surreal/kube/surreal_kube.py:214
↓ 1 callersMethod_gae_and_return
computes generalized advantage estimate and corresponding N-step return. Details of algorithm can be found here: https://arxiv.org/p
surreal/learner/ppo.py:355
↓ 1 callersMethod_gather_action_infos
Gathers corresponding action informations from partial trajectories Args: experience: (type: dictionary) subt
surreal/learner/aggregator.py:223
↓ 1 callersMethod_gcloud_nfs_exec
(self, command)
surreal/kube/surreal_kube.py:539
↓ 1 callersFunction_get_bound_args
https://docs.python.org/3/library/inspect.html#inspect.BoundArguments def f(a, b, c=5, d=6): pass get_bound_args(f, 3, 6, d=100) -> {'a':
surreal/utils/common.py:174
↓ 1 callersMethod_get_tag
(self, tag)
surreal/env/monitor.py:142
↓ 1 callersMethod_get_tag
(self, tag)
surreal/env/monitor.py:197
↓ 1 callersMethod_get_tensorplex
Get the periodic tensorplex object Args: @name: The name of the collection of metrics
surreal/learner/base.py:177
↓ 1 callersMethod_get_video_for_experiment
(self, experiment_name, save_folder,
surreal/kube/surreal_kube.py:476
↓ 1 callersFunction_has_req
(config)
surreal/session/config.py:60
↓ 1 callersMethod_hash_nested
(self, values)
surreal/distributed/exp_sender.py:45
↓ 1 callersMethod_init_noise
initializes exploration noise and populates self.noise, a callable that returns noise of dimension same as action
surreal/agent/ddpg_agent.py:115
↓ 1 callersMethod_initialize
implements AutoInitializeMeta meta class. self.module_dict can only happen after the module is constructed by subclasses.
surreal/agent/base.py:66
↓ 1 callersMethod_optimize
Note that while the replay contains uint8, the aggregator returns float32 tensors Arguments: obs: an observation
surreal/learner/ddpg.py:244
↓ 1 callersMethod_optimize
main method for optimization that calls _adapt/clip_update and _value_update epoch_policy and epoch_baseline times respectiv
surreal/learner/ppo.py:487
↓ 1 callersMethod_optimize
(self, obs, actions, rewards, obs_next, dones, weights)
surreal/learner/dqn.py:42
↓ 1 callersMethod_post_publish
function that manages metrics and behavior after parameter release Actions include: adjusts adaptive thresho
surreal/learner/ppo.py:637
↓ 1 callersMethod_preprocess_batch_ppo
Loading experiences from numpy to torch.FloatTensor type Args: batch: BeneDict of experiences containing fol
surreal/learner/ppo.py:420
↓ 1 callersMethod_process_experiment_name
experiment_name will be used as DNS, so must not have underscore or dot
surreal/tmux/surreal_tmux.py:103
↓ 1 callersMethod_reduce_helper
(self, start, end, node, node_start, node_end)
surreal/replay/segment_tree.py:35
↓ 1 callersMethod_render
(self, *args, **kwargs)
surreal/env/base.py:75
↓ 1 callersMethod_reset
(self)
surreal/env/base.py:72
↓ 1 callersMethod_retrieve_storage
Args: exp: a nested dict or list Only dict keys that end with `_hash` will be retrieved. The proc
surreal/distributed/exp_collector.py:37
↓ 1 callersMethod_reverse_action
(self, action)
surreal/env/wrapper.py:139
↓ 1 callersMethod_reward
(self, reward)
surreal/env/wrapper.py:121
↓ 1 callersFunction_run_cmd_list
(args)
docker/build_files/entrypoint-ubuntu.py:16
↓ 1 callersFunction_run_cmd_list
(args)
docker/build_files/entrypoint-nvidia.py:17
↓ 1 callersMethod_run_optimizer
(self, loss)
surreal/learner/dqn.py:30
↓ 1 callersMethod_sample_proportional
This is a helper function to sample expriences with probabilities proportional to their priorities. Returns a list of indice
surreal/replay/prioritized_replay.py:111
↓ 1 callersMethod_save_ckpt
(self, suffix)
surreal/utils/checkpoint.py:234
↓ 1 callersMethod_save_metadata
(self)
surreal/utils/checkpoint.py:231
↓ 1 callersMethod_setup_checkpoint
(self)
surreal/learner/base.py:260
↓ 1 callersMethod_setup_create
(self)
surreal/tmux/surreal_tmux.py:58
↓ 1 callersMethod_setup_create
(self)
surreal/kube/surreal_kube.py:114
↓ 1 callersMethod_setup_docker_clean
(self)
surreal/kube/surreal_kube.py:101
↓ 1 callersFunction_setup_env
Setup the necessary environment variables
surreal/test_helpers/integration_test.py:14
↓ 1 callersMethod_setup_get_config
(self)
surreal/kube/surreal_kube.py:82
↓ 1 callersMethod_setup_get_tensorboard
(self)
surreal/kube/surreal_kube.py:90
↓ 1 callersMethod_setup_get_videos
(self)
surreal/kube/surreal_kube.py:72
↓ 1 callersMethod_setup_gpu
Assigns GPU to agents and learners in an optimal way. No GPU, do nothing
surreal/subproc/surreal_subproc.py:203
↓ 1 callersMethod_setup_gpu
Assigns GPU to agents and learners in an optimal way. No GPU, do nothing
surreal/tmux/surreal_tmux.py:210
↓ 1 callersMethod_setup_logging
(self)
surreal/replay/base.py:112
↓ 1 callersMethod_setup_logging
(self)
surreal/learner/base.py:159
↓ 1 callersMethod_setup_logging
Creates tensorplex logger and loggerplex logger Initializes bookkeeping values
surreal/agent/base.py:86
↓ 1 callersMethod_setup_parameter_pull
(self)
surreal/agent/base.py:81
↓ 1 callersMethod_setup_prefetching
(self)
surreal/learner/base.py:102
↓ 1 callersMethod_setup_publish
(self)
surreal/learner/base.py:90
↓ 1 callersMethod_setup_tensorboard
(self)
surreal/kube/surreal_kube.py:104
← previousnext →101–200 of 791, ranked by callers