MCPcopy Create free account

hub / github.com/AIcrowd/neurips2020-procgen-starter-kit / functions

Functions76 in github.com/AIcrowd/neurips2020-procgen-starter-kit

↓ 3 callersFunctionconv_layer
(depth, name)
models/impala_cnn_tf.py:8
↓ 2 callersMethod__init__
(self, obs_space, action_space, num_outputs, model_config, name)
models/impala_cnn_torch.py:54
↓ 2 callersMethod_get_tmp_progress_filename
(self)
rollout.py:96
↓ 2 callersFunction_source_file
Dynamically "sources" a provided file
utils/loader.py:13
↓ 2 callersFunctionconv_layers
(x, model_config, obs_space, prefix="")
models/my_vision_network.py:33
↓ 2 callersFunctionfc_layers
(x, model_config, prefix="")
models/my_vision_network.py:53
↓ 2 callersFunctionget_num_workers
(ctx)
utils/validate_config.py:25
↓ 2 callersMethodget_output_shape
(self)
models/impala_cnn_torch.py:41
↓ 2 callersFunctionkeep_going
Determine whether we've collected enough data
rollout.py:337
↓ 2 callersFunctionload_algorithms
This function loads the custom algorithms implemented in this repository, and registers them with the tune registry
utils/loader.py:85
↓ 2 callersFunctionload_envs
This function takes a path to a local directory and looks for an `envs` folder, and imports all the available files in there.
utils/loader.py:45
↓ 2 callersFunctionload_models
This function takes a path to a local directory and looks for a `models` folder, and imports all the available files in there.
utils/loader.py:65
↓ 2 callersFunctionload_preprocessors
Function to register custom preprocessors
utils/loader.py:99
↓ 2 callersMethodreset
(self)
envs/procgen_env_wrapper.py:46
↓ 2 callersFunctionresidual_block
(x, depth, prefix)
models/impala_cnn_tf.py:14
↓ 2 callersMethodstep
(self, action)
envs/procgen_env_wrapper.py:50
↓ 1 callersMethod_get_progress
(self)
rollout.py:140
↓ 1 callersMethodappend_step
Add a step to the current rollout, if we are saving them
rollout.py:168
↓ 1 callersMethodbegin_rollout
(self)
rollout.py:150
↓ 1 callersFunctioncheck_cpu_allotment
(ctx)
utils/validate_config.py:33
↓ 1 callersFunctioncheck_gpu_allotment
(ctx)
utils/validate_config.py:37
↓ 1 callersMethodclose
(self)
envs/procgen_env_wrapper.py:58
↓ 1 callersFunctionconv_sequence
(x, depth, prefix)
models/impala_cnn_tf.py:24
↓ 1 callersFunctioncreate_parser
(parser_creator=None)
train.py:52
↓ 1 callersFunctioncreate_parser
(parser_creator=None)
rollout.py:180
↓ 1 callersMethodend_rollout
(self)
rollout.py:153
↓ 1 callersFunctionget_conv_activation
(model_config)
models/my_vision_network.py:18
↓ 1 callersFunctionget_experiment_config_file
()
utils/validate_config.py:8
↓ 1 callersFunctionget_fc_activation
(model_config)
models/my_vision_network.py:26
↓ 1 callersFunctionget_final_fc
(x, num_outputs, model_config)
models/my_vision_network.py:64
↓ 1 callersFunctionmain
()
utils/validate_config.py:47
↓ 1 callersFunctionread_yaml
(fpath)
utils/validate_config.py:19
↓ 1 callersMethodrender
(self, mode="human")
envs/procgen_env_wrapper.py:55
↓ 1 callersFunctionrollout
(agent, env_name, num_steps, num_episodes=0, saver=None,
rollout.py:349
↓ 1 callersFunctionrun
(args, parser)
train.py:151
↓ 1 callersFunctionrun
(args, parser)
rollout.py:257
↓ 1 callersFunctionvalue_layers
(x, inputs, obs_space, model_config)
models/my_vision_network.py:69
Method__enter__
(self)
rollout.py:104
Method__exit__
(self, type, value, traceback)
rollout.py:127
Method__init__
(self, outfile=None, use_shelve=False, write_update_file=Fa
rollout.py:76
Method__init__
(self, config)
envs/procgen_env_wrapper.py:12
Method__init__
(self, obs_space, action_space, num_outputs, model_config, name)
models/impala_cnn_tf.py:40
Method__init__
(self, obs_space, action_space, num_outputs, model_config, name)
models/my_vision_network.py:80
Method__init__
(self, channels)
models/impala_cnn_torch.py:10
Method__init__
(self, input_shape, out_channels)
models/impala_cnn_torch.py:25
Method__init__
(self, observation_space, action_space, config)
algorithms/random_policy/policy.py:17
Method__missing__
(self, key)
rollout.py:328
Method__repr__
(self)
envs/procgen_env_wrapper.py:64
Function_import_custom_random_agent
()
algorithms/registry.py:16
Function_import_random_policy
()
algorithms/registry.py:20
Method_init
(self, config, env_creator)
algorithms/custom_random_agent/custom_random_agent.py:22
Method_init_shape
(self, obs_space, options)
preprocessors/custom_preprocessor.py:12
Method_train
(self)
algorithms/custom_random_agent/custom_random_agent.py:26
Methodcompute_actions
Return the action for a batch Returns: action_batch: List of actions for the batch rnn_states: List of RNN states if
algorithms/random_policy/policy.py:26
Functiondefault_policy_agent_mapping
(unused_agent_id)
rollout.py:333
Methodforward
(self, input_dict, state, seq_lens)
models/impala_cnn_tf.py:60
Methodforward
(self, input_dict, state, seq_lens)
models/my_vision_network.py:99
Methodforward
(self, x)
models/impala_cnn_torch.py:15
Methodforward
(self, x)
models/impala_cnn_torch.py:33
Methodforward
(self, input_dict, state, seq_lens)
models/impala_cnn_torch.py:74
Methodget_weights
Returns model weights. Returns: weights (obj): Serializable copy or view of model weights
algorithms/random_policy/policy.py:67
Methodlearn_on_batch
Fused compute gradients and apply gradients call. Either this or the combination of compute/apply grads must be implemented by subcla
algorithms/random_policy/policy.py:50
Methodon_episode_end
Runs when an episode is done. Args: worker (RolloutWorker): Reference to the current rollout worker. base_env (BaseEnv
callbacks.py:59
Methodon_episode_start
Callback run on the rollout worker before each episode starts. Args: worker (RolloutWorker): Reference to the current rollout work
callbacks.py:26
Methodon_episode_step
Runs on each episode step. Args: worker (RolloutWorker): Reference to the current rollout worker. base_env (BaseEnv):
callbacks.py:44
Methodon_postprocess_trajectory
Called immediately after a policy's postprocess_fn is called. You can use this callback to do additional postprocessing for a policy,
callbacks.py:84
Methodon_sample_end
Called at the end RolloutWorker.sample(). Args: worker (RolloutWorker): Reference to the current rollout worker. sampl
callbacks.py:109
Methodon_train_result
Called at the end of Trainable.train(). Args: trainer (Trainer): Current trainer instance. result (dict): Dict of resu
callbacks.py:120
Methodoutfile
(self)
rollout.py:101
Methodseed
(self, seed=None)
envs/procgen_env_wrapper.py:61
Methodset_weights
Returns the current exploration information of this policy. This information depends on the policy's Exploration object. Ret
algorithms/random_policy/policy.py:75
Methodspec
(self)
envs/procgen_env_wrapper.py:68
Methodtransform
(self, observation)
preprocessors/custom_preprocessor.py:15
Methodvalue_function
(self)
models/impala_cnn_tf.py:66
Methodvalue_function
(self)
models/my_vision_network.py:106
Methodvalue_function
(self)
models/impala_cnn_torch.py:90