Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/AIcrowd/neurips2020-procgen-starter-kit
/ functions
Functions
76 in github.com/AIcrowd/neurips2020-procgen-starter-kit
⨍
Functions
76
◇
Types & classes
12
↓ 3 callers
Function
conv_layer
(depth, name)
models/impala_cnn_tf.py:8
↓ 2 callers
Method
__init__
(self, obs_space, action_space, num_outputs, model_config, name)
models/impala_cnn_torch.py:54
↓ 2 callers
Method
_get_tmp_progress_filename
(self)
rollout.py:96
↓ 2 callers
Function
_source_file
Dynamically "sources" a provided file
utils/loader.py:13
↓ 2 callers
Function
conv_layers
(x, model_config, obs_space, prefix="")
models/my_vision_network.py:33
↓ 2 callers
Function
fc_layers
(x, model_config, prefix="")
models/my_vision_network.py:53
↓ 2 callers
Function
get_num_workers
(ctx)
utils/validate_config.py:25
↓ 2 callers
Method
get_output_shape
(self)
models/impala_cnn_torch.py:41
↓ 2 callers
Function
keep_going
Determine whether we've collected enough data
rollout.py:337
↓ 2 callers
Function
load_algorithms
This function loads the custom algorithms implemented in this repository, and registers them with the tune registry
utils/loader.py:85
↓ 2 callers
Function
load_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 callers
Function
load_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 callers
Function
load_preprocessors
Function to register custom preprocessors
utils/loader.py:99
↓ 2 callers
Method
reset
(self)
envs/procgen_env_wrapper.py:46
↓ 2 callers
Function
residual_block
(x, depth, prefix)
models/impala_cnn_tf.py:14
↓ 2 callers
Method
step
(self, action)
envs/procgen_env_wrapper.py:50
↓ 1 callers
Method
_get_progress
(self)
rollout.py:140
↓ 1 callers
Method
append_step
Add a step to the current rollout, if we are saving them
rollout.py:168
↓ 1 callers
Method
begin_rollout
(self)
rollout.py:150
↓ 1 callers
Function
check_cpu_allotment
(ctx)
utils/validate_config.py:33
↓ 1 callers
Function
check_gpu_allotment
(ctx)
utils/validate_config.py:37
↓ 1 callers
Method
close
(self)
envs/procgen_env_wrapper.py:58
↓ 1 callers
Function
conv_sequence
(x, depth, prefix)
models/impala_cnn_tf.py:24
↓ 1 callers
Function
create_parser
(parser_creator=None)
train.py:52
↓ 1 callers
Function
create_parser
(parser_creator=None)
rollout.py:180
↓ 1 callers
Method
end_rollout
(self)
rollout.py:153
↓ 1 callers
Function
get_conv_activation
(model_config)
models/my_vision_network.py:18
↓ 1 callers
Function
get_experiment_config_file
()
utils/validate_config.py:8
↓ 1 callers
Function
get_fc_activation
(model_config)
models/my_vision_network.py:26
↓ 1 callers
Function
get_final_fc
(x, num_outputs, model_config)
models/my_vision_network.py:64
↓ 1 callers
Function
main
()
utils/validate_config.py:47
↓ 1 callers
Function
read_yaml
(fpath)
utils/validate_config.py:19
↓ 1 callers
Method
render
(self, mode="human")
envs/procgen_env_wrapper.py:55
↓ 1 callers
Function
rollout
(agent, env_name, num_steps, num_episodes=0, saver=None,
rollout.py:349
↓ 1 callers
Function
run
(args, parser)
train.py:151
↓ 1 callers
Function
run
(args, parser)
rollout.py:257
↓ 1 callers
Function
value_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
Method
compute_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
Function
default_policy_agent_mapping
(unused_agent_id)
rollout.py:333
Method
forward
(self, input_dict, state, seq_lens)
models/impala_cnn_tf.py:60
Method
forward
(self, input_dict, state, seq_lens)
models/my_vision_network.py:99
Method
forward
(self, x)
models/impala_cnn_torch.py:15
Method
forward
(self, x)
models/impala_cnn_torch.py:33
Method
forward
(self, input_dict, state, seq_lens)
models/impala_cnn_torch.py:74
Method
get_weights
Returns model weights. Returns: weights (obj): Serializable copy or view of model weights
algorithms/random_policy/policy.py:67
Method
learn_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
Method
on_episode_end
Runs when an episode is done. Args: worker (RolloutWorker): Reference to the current rollout worker. base_env (BaseEnv
callbacks.py:59
Method
on_episode_start
Callback run on the rollout worker before each episode starts. Args: worker (RolloutWorker): Reference to the current rollout work
callbacks.py:26
Method
on_episode_step
Runs on each episode step. Args: worker (RolloutWorker): Reference to the current rollout worker. base_env (BaseEnv):
callbacks.py:44
Method
on_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
Method
on_sample_end
Called at the end RolloutWorker.sample(). Args: worker (RolloutWorker): Reference to the current rollout worker. sampl
callbacks.py:109
Method
on_train_result
Called at the end of Trainable.train(). Args: trainer (Trainer): Current trainer instance. result (dict): Dict of resu
callbacks.py:120
Method
outfile
(self)
rollout.py:101
Method
seed
(self, seed=None)
envs/procgen_env_wrapper.py:61
Method
set_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
Method
spec
(self)
envs/procgen_env_wrapper.py:68
Method
transform
(self, observation)
preprocessors/custom_preprocessor.py:15
Method
value_function
(self)
models/impala_cnn_tf.py:66
Method
value_function
(self)
models/my_vision_network.py:106
Method
value_function
(self)
models/impala_cnn_torch.py:90