id: 1 unit: def main() file: run_inverse_dynamics_model.py start line: 0 end line: 0 size: 58 LOC McCabe index: 7 number of parameters: 6 id: 2 unit: def behavioural_cloning_train() file: behavioural_cloning.py start line: 0 end line: 0 size: 54 LOC McCabe index: 8 number of parameters: 4 id: 3 unit: def data_loader_worker() file: data_loader.py start line: 0 end line: 0 size: 53 LOC McCabe index: 22 number of parameters: 3 id: 4 unit: def __init__() file: data_loader.py start line: 0 end line: 0 size: 42 LOC McCabe index: 12 number of parameters: 6 id: 5 unit: def _preproc() file: lib/xf.py start line: 0 end line: 0 size: 40 LOC McCabe index: 5 number of parameters: 5 id: 6 unit: def _infer_part() file: lib/misc.py start line: 0 end line: 0 size: 38 LOC McCabe index: 13 number of parameters: 5 id: 7 unit: def json_action_to_env_action() file: run_inverse_dynamics_model.py start line: 0 end line: 0 size: 31 LOC McCabe index: 12 number of parameters: 1 id: 8 unit: def __init__() file: lib/policy.py start line: 0 end line: 0 size: 29 LOC McCabe index: 4 number of parameters: 6 id: 9 unit: def from_factored() file: lib/action_mapping.py start line: 0 end line: 0 size: 27 LOC McCabe index: 10 number of parameters: 2 id: 10 unit: def parse_dtype() file: lib/torch_util.py start line: 0 end line: 0 size: 26 LOC McCabe index: 18 number of parameters: 1 id: 11 unit: def preproc_qkv() file: lib/xf.py start line: 0 end line: 0 size: 25 LOC McCabe index: 10 number of parameters: 4 id: 12 unit: def reshape_undo() file: lib/misc.py start line: 0 end line: 0 size: 25 LOC McCabe index: 5 number of parameters: 7 id: 13 unit: def residual() file: lib/xf.py start line: 0 end line: 0 size: 23 LOC McCabe index: 3 number of parameters: 3 id: 14 unit: def _precompute_to_factored() file: lib/action_mapping.py start line: 0 end line: 0 size: 21 LOC McCabe index: 7 number of parameters: 1 id: 15 unit: def forward() file: lib/policy.py start line: 0 end line: 0 size: 20 LOC McCabe index: 5 number of parameters: 4 id: 16 unit: def prefix_multimap() file: lib/tree_util.py start line: 0 end line: 0 size: 19 LOC McCabe index: 7 number of parameters: 4 id: 17 unit: def __init__() file: agent.py start line: 0 end line: 0 size: 18 LOC McCabe index: 5 number of parameters: 5 id: 18 unit: def forward() file: lib/util.py start line: 0 end line: 0 size: 18 LOC McCabe index: 6 number of parameters: 4 id: 19 unit: def _handle_ellipsis() file: lib/misc.py start line: 0 end line: 0 size: 18 LOC McCabe index: 4 number of parameters: 3 id: 20 unit: def get_mask() file: lib/masked_attention.py start line: 0 end line: 0 size: 18 LOC McCabe index: 3 number of parameters: 7 id: 21 unit: def store_args() file: lib/minecraft_util.py start line: 0 end line: 0 size: 18 LOC McCabe index: 4 number of parameters: 1 id: 22 unit: def _infer_question_mark() file: lib/misc.py start line: 0 end line: 0 size: 17 LOC McCabe index: 5 number of parameters: 2 id: 23 unit: def factored_buttons_to_groups() file: lib/action_mapping.py start line: 0 end line: 0 size: 16 LOC McCabe index: 13 number of parameters: 3 id: 24 unit: def __init__() file: lib/action_mapping.py start line: 0 end line: 0 size: 16 LOC McCabe index: 6 number of parameters: 3 id: 25 unit: def forward() file: lib/masked_attention.py start line: 0 end line: 0 size: 16 LOC McCabe index: 2 number of parameters: 4 id: 26 unit: def forward() file: lib/normalize_ewma.py start line: 0 end line: 0 size: 16 LOC McCabe index: 3 number of parameters: 2 id: 27 unit: def get_norm_entropy_from_cat_head() file: lib/minecraft_util.py start line: 0 end line: 0 size: 16 LOC McCabe index: 5 number of parameters: 4 id: 28 unit: def tree_multimap() file: lib/tree_util.py start line: 0 end line: 0 size: 16 LOC McCabe index: 7 number of parameters: 4 id: 29 unit: def main() file: run_agent.py start line: 0 end line: 0 size: 15 LOC McCabe index: 2 number of parameters: 2 id: 30 unit: def __init__() file: lib/util.py start line: 0 end line: 0 size: 15 LOC McCabe index: 1 number of parameters: 0 id: 31 unit: def recurrent_forward() file: lib/util.py start line: 0 end line: 0 size: 15 LOC McCabe index: 5 number of parameters: 5 id: 32 unit: def dict_to_numpy() file: lib/actions.py start line: 0 end line: 0 size: 15 LOC McCabe index: 4 number of parameters: 2 id: 33 unit: def act() file: lib/policy.py start line: 0 end line: 0 size: 15 LOC McCabe index: 3 number of parameters: 7 id: 34 unit: def __next__() file: data_loader.py start line: 0 end line: 0 size: 14 LOC McCabe index: 3 number of parameters: 1 id: 35 unit: def __init__() file: lib/util.py start line: 0 end line: 0 size: 14 LOC McCabe index: 1 number of parameters: 0 id: 36 unit: def __init__() file: lib/xf.py start line: 0 end line: 0 size: 14 LOC McCabe index: 1 number of parameters: 7 id: 37 unit: def _parse_reshape_str() file: lib/misc.py start line: 0 end line: 0 size: 14 LOC McCabe index: 6 number of parameters: 2 id: 38 unit: def __init__() file: lib/xf.py start line: 0 end line: 0 size: 13 LOC McCabe index: 1 number of parameters: 0 id: 39 unit: def forward() file: lib/xf.py start line: 0 end line: 0 size: 13 LOC McCabe index: 2 number of parameters: 2 id: 40 unit: def get_norm_cat_entropy() file: lib/minecraft_util.py start line: 0 end line: 0 size: 13 LOC McCabe index: 4 number of parameters: 4 id: 41 unit: def __init__() file: inverse_dynamics_model.py start line: 0 end line: 0 size: 13 LOC McCabe index: 3 number of parameters: 4 id: 42 unit: def validate_env() file: agent.py start line: 0 end line: 0 size: 12 LOC McCabe index: 7 number of parameters: 1 id: 43 unit: def __init__() file: lib/xf.py start line: 0 end line: 0 size: 12 LOC McCabe index: 1 number of parameters: 0 id: 44 unit: def update_state() file: lib/xf.py start line: 0 end line: 0 size: 12 LOC McCabe index: 2 number of parameters: 4 id: 45 unit: def __init__() file: lib/impala_cnn.py start line: 0 end line: 0 size: 12 LOC McCabe index: 1 number of parameters: 0 id: 46 unit: def forward() file: lib/policy.py start line: 0 end line: 0 size: 12 LOC McCabe index: 3 number of parameters: 4 id: 47 unit: def NormedLinear() file: lib/torch_util.py start line: 0 end line: 0 size: 12 LOC McCabe index: 4 number of parameters: 4 id: 48 unit: def make_action_head() file: lib/action_head.py start line: 0 end line: 0 size: 12 LOC McCabe index: 7 number of parameters: 3 id: 49 unit: def get_diag_guassian_entropy() file: lib/minecraft_util.py start line: 0 end line: 0 size: 12 LOC McCabe index: 4 number of parameters: 3 id: 50 unit: def _env_action_to_agent() file: agent.py start line: 0 end line: 0 size: 11 LOC McCabe index: 9 number of parameters: 4 id: 51 unit: def _infer_step() file: lib/misc.py start line: 0 end line: 0 size: 11 LOC McCabe index: 3 number of parameters: 1 id: 52 unit: def load_average_with_metadata() file: lib/torch_util.py start line: 0 end line: 0 size: 11 LOC McCabe index: 4 number of parameters: 2 id: 53 unit: def index() file: lib/torch_util.py start line: 0 end line: 0 size: 11 LOC McCabe index: 2 number of parameters: 2 id: 54 unit: def __init__() file: lib/mlp.py start line: 0 end line: 0 size: 11 LOC McCabe index: 2 number of parameters: 7 id: 55 unit: def __init__() file: lib/masked_attention.py start line: 0 end line: 0 size: 11 LOC McCabe index: 1 number of parameters: 0 id: 56 unit: def walk_pytree() file: lib/tree_util.py start line: 0 end line: 0 size: 11 LOC McCabe index: 5 number of parameters: 4 id: 57 unit: def _agent_action_to_env() file: agent.py start line: 0 end line: 0 size: 10 LOC McCabe index: 2 number of parameters: 2 id: 58 unit: def attention() file: lib/xf.py start line: 0 end line: 0 size: 10 LOC McCabe index: 1 number of parameters: 0 id: 59 unit: def __init__() file: lib/action_mapping.py start line: 0 end line: 0 size: 10 LOC McCabe index: 1 number of parameters: 2 id: 60 unit: def __init__() file: lib/impala_cnn.py start line: 0 end line: 0 size: 10 LOC McCabe index: 1 number of parameters: 0 id: 61 unit: def forward() file: lib/policy.py start line: 0 end line: 0 size: 10 LOC McCabe index: 3 number of parameters: 4 id: 62 unit: def LayerNorm() file: lib/torch_util.py start line: 0 end line: 0 size: 10 LOC McCabe index: 3 number of parameters: 3 id: 63 unit: def forward() file: lib/action_head.py start line: 0 end line: 0 size: 10 LOC McCabe index: 4 number of parameters: 3 id: 64 unit: def forward() file: lib/action_head.py start line: 0 end line: 0 size: 10 LOC McCabe index: 5 number of parameters: 3 id: 65 unit: def __init__() file: lib/normalize_ewma.py start line: 0 end line: 0 size: 10 LOC McCabe index: 1 number of parameters: 6 id: 66 unit: def tree_map() file: lib/tree_util.py start line: 0 end line: 0 size: 10 LOC McCabe index: 5 number of parameters: 3 id: 67 unit: def forward() file: lib/util.py start line: 0 end line: 0 size: 9 LOC McCabe index: 2 number of parameters: 4 id: 68 unit: def bandify() file: lib/util.py start line: 0 end line: 0 size: 9 LOC McCabe index: 2 number of parameters: 3 id: 69 unit: def act() file: lib/xf.py start line: 0 end line: 0 size: 9 LOC McCabe index: 4 number of parameters: 2 id: 70 unit: def _is_separate() file: lib/xf.py start line: 0 end line: 0 size: 9 LOC McCabe index: 3 number of parameters: 2 id: 71 unit: def forward() file: lib/impala_cnn.py start line: 0 end line: 0 size: 9 LOC McCabe index: 1 number of parameters: 2 id: 72 unit: def compose_undo() file: lib/misc.py start line: 0 end line: 0 size: 9 LOC McCabe index: 2 number of parameters: 2 id: 73 unit: def fixed_point() file: lib/misc.py start line: 0 end line: 0 size: 9 LOC McCabe index: 4 number of parameters: 3 id: 74 unit: def __init__() file: lib/policy.py start line: 0 end line: 0 size: 9 LOC McCabe index: 3 number of parameters: 3 id: 75 unit: def forward() file: lib/policy.py start line: 0 end line: 0 size: 9 LOC McCabe index: 3 number of parameters: 5 id: 76 unit: def save_kwargs() file: lib/torch_util.py start line: 0 end line: 0 size: 9 LOC McCabe index: 2 number of parameters: 1 id: 77 unit: def kl_divergence() file: lib/action_head.py start line: 0 end line: 0 size: 9 LOC McCabe index: 1 number of parameters: 3 id: 78 unit: def predict_actions() file: inverse_dynamics_model.py start line: 0 end line: 0 size: 9 LOC McCabe index: 2 number of parameters: 2 id: 79 unit: def get_action() file: agent.py start line: 0 end line: 0 size: 8 LOC McCabe index: 2 number of parameters: 2 id: 80 unit: def discretize() file: lib/actions.py start line: 0 end line: 0 size: 8 LOC McCabe index: 2 number of parameters: 2 id: 81 unit: def undiscretize() file: lib/actions.py start line: 0 end line: 0 size: 8 LOC McCabe index: 2 number of parameters: 2 id: 82 unit: def env2policy() file: lib/actions.py start line: 0 end line: 0 size: 8 LOC McCabe index: 2 number of parameters: 2 id: 83 unit: def split_heads() file: lib/xf.py start line: 0 end line: 0 size: 8 LOC McCabe index: 1 number of parameters: 2 id: 84 unit: def to_factored() file: lib/action_mapping.py start line: 0 end line: 0 size: 8 LOC McCabe index: 1 number of parameters: 2 id: 85 unit: def forward() file: lib/impala_cnn.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 2 id: 86 unit: def _conv3d_forward() file: lib/policy.py start line: 0 end line: 0 size: 8 LOC McCabe index: 2 number of parameters: 2 id: 87 unit: def contextmanager_to_decorator() file: lib/torch_util.py start line: 0 end line: 0 size: 8 LOC McCabe index: 1 number of parameters: 1 id: 88 unit: def get_band_diagonal_mask() file: lib/masked_attention.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 5 id: 89 unit: def logprob() file: lib/action_head.py start line: 0 end line: 0 size: 8 LOC McCabe index: 2 number of parameters: 3 id: 90 unit: def _agent_action_to_env() file: inverse_dynamics_model.py start line: 0 end line: 0 size: 8 LOC McCabe index: 1 number of parameters: 2 id: 91 unit: def composite_images_with_alpha() file: data_loader.py start line: 0 end line: 0 size: 7 LOC McCabe index: 3 number of parameters: 5 id: 92 unit: def get_module_log_keys_recursive() file: lib/util.py start line: 0 end line: 0 size: 7 LOC McCabe index: 3 number of parameters: 1 id: 93 unit: def forward() file: lib/util.py start line: 0 end line: 0 size: 7 LOC McCabe index: 3 number of parameters: 2 id: 94 unit: def get_norm() file: lib/util.py start line: 0 end line: 0 size: 7 LOC McCabe index: 3 number of parameters: 3 id: 95 unit: def numpy_to_dict() file: lib/actions.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 2 id: 96 unit: def preproc_qkv() file: lib/xf.py start line: 0 end line: 0 size: 7 LOC McCabe index: 4 number of parameters: 2 id: 97 unit: def postproc_a() file: lib/xf.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 3 id: 98 unit: def relattn_logits() file: lib/xf.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 3 id: 99 unit: def __init__() file: lib/impala_cnn.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 0 id: 100 unit: def output_shape() file: lib/impala_cnn.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 2 id: 101 unit: def _ground() file: lib/misc.py start line: 0 end line: 0 size: 7 LOC McCabe index: 3 number of parameters: 3 id: 102 unit: def forward() file: lib/policy.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 2 id: 103 unit: def __init__() file: lib/policy.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 0 id: 104 unit: def forward() file: lib/mlp.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 2 id: 105 unit: def sample() file: lib/action_head.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 3 id: 106 unit: def sample() file: lib/action_head.py start line: 0 end line: 0 size: 7 LOC McCabe index: 3 number of parameters: 3 id: 107 unit: def unzip2() file: lib/tree_util.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 1 id: 108 unit: def __init__() file: lib/util.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 0 id: 109 unit: def __init__() file: lib/actions.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 0 id: 110 unit: def make_maybe_multiscale() file: lib/xf.py start line: 0 end line: 0 size: 6 LOC McCabe index: 3 number of parameters: 6 id: 111 unit: def safezip() file: lib/misc.py start line: 0 end line: 0 size: 6 LOC McCabe index: 3 number of parameters: 1 id: 112 unit: def transpose() file: lib/misc.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 3 id: 113 unit: def __init__() file: lib/policy.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 4 id: 114 unit: def initial_state() file: lib/masked_attention.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 3 id: 115 unit: def __init__() file: lib/action_head.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 3 id: 116 unit: def forward() file: lib/action_head.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 3 id: 117 unit: def logprob() file: lib/action_head.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 3 id: 118 unit: def entropy() file: lib/action_head.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 2 id: 119 unit: def safe_map() file: lib/tree_util.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 2 id: 120 unit: def build_tree() file: lib/tree_util.py start line: 0 end line: 0 size: 6 LOC McCabe index: 3 number of parameters: 2 id: 121 unit: def _tree_unflatten() file: lib/tree_util.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 2 id: 122 unit: def _nested_treedef() file: lib/tree_util.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 2 id: 123 unit: def __repr__() file: lib/tree_util.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 1 id: 124 unit: def load_model_parameters() file: behavioural_cloning.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 1 id: 125 unit: def get_log_keys() file: lib/util.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 1 id: 126 unit: def initial_state() file: lib/util.py start line: 0 end line: 0 size: 5 LOC McCabe index: 4 number of parameters: 2 id: 127 unit: def _banded_repeat() file: lib/util.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 2 id: 128 unit: def __init__() file: lib/xf.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 5 id: 129 unit: def _required_padding() file: lib/xf.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 2 id: 130 unit: def initial_state() file: lib/xf.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 3 id: 131 unit: def get_action_space_update() file: lib/action_mapping.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 1 id: 132 unit: def get_action_space_update() file: lib/action_mapping.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 1 id: 133 unit: def intprod() file: lib/misc.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 1 id: 134 unit: def transpose_undo() file: lib/misc.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 5 id: 135 unit: def _infer() file: lib/misc.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 3 id: 136 unit: def product() file: lib/misc.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 2 id: 137 unit: def initial_state() file: lib/policy.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 2 id: 138 unit: def reset_parameters() file: lib/policy.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 1 id: 139 unit: def get_logprob_of_action() file: lib/policy.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 3 id: 140 unit: def get_output_for_observation() file: lib/policy.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 4 id: 141 unit: def v() file: lib/policy.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 4 id: 142 unit: def __init__() file: lib/policy.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 0 id: 143 unit: def __init__() file: lib/policy.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 0 id: 144 unit: def predict() file: lib/policy.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 0 id: 145 unit: def reset_parameters() file: lib/action_head.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 1 id: 146 unit: def kl_divergence() file: lib/action_head.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 3 id: 147 unit: def running_mean_var() file: lib/normalize_ewma.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 1 id: 148 unit: def partial() file: lib/tree_util.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 3 id: 149 unit: def safe_zip() file: lib/tree_util.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 1 id: 150 unit: def __eq__() file: lib/tree_util.py start line: 0 end line: 0 size: 5 LOC McCabe index: 4 number of parameters: 2 id: 151 unit: def _video_obs_to_agent() file: inverse_dynamics_model.py start line: 0 end line: 0 size: 5 LOC McCabe index: 3 number of parameters: 2 id: 152 unit: def _env_obs_to_agent() file: agent.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 153 unit: def __del__() file: data_loader.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 1 id: 154 unit: def __init__() file: lib/xf.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 5 id: 155 unit: def residual() file: lib/xf.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 156 unit: def __init__() file: lib/xf.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 3 id: 157 unit: def reset_parameters() file: lib/policy.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 1 id: 158 unit: def reset_parameters() file: lib/scaled_mse_head.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 1 id: 159 unit: def sequential() file: lib/torch_util.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 5 id: 160 unit: def fan_in_linear() file: lib/action_head.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 3 id: 161 unit: def reset_parameters() file: lib/normalize_ewma.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 1 id: 162 unit: def __init__() file: lib/tree_util.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 4 id: 163 unit: def __init__() file: lib/tree_util.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 4 id: 164 unit: def resize_image() file: agent.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 165 unit: def load_weights() file: agent.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 166 unit: def policy2env() file: lib/actions.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 167 unit: def __init__() file: lib/xf.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 168 unit: def preproc_r() file: lib/xf.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 169 unit: def preproc_r() file: lib/xf.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 170 unit: def forward() file: lib/xf.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 171 unit: def stateless_forward() file: lib/xf.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 172 unit: def forward() file: lib/impala_cnn.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 173 unit: def reshape() file: lib/misc.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 174 unit: def exact_div() file: lib/misc.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 175 unit: def set_default_torch_device() file: lib/torch_util.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 176 unit: def flatten_image() file: lib/torch_util.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 177 unit: def reset_parameters() file: lib/action_head.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 178 unit: def entropy() file: lib/action_head.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 179 unit: def reset_parameters() file: lib/action_head.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 1 id: 180 unit: def denormalize() file: lib/normalize_ewma.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 181 unit: def register_pytree_node() file: lib/tree_util.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 182 unit: def dict_to_iterable() file: lib/tree_util.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 183 unit: def load_weights() file: inverse_dynamics_model.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 184 unit: def reset() file: agent.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 185 unit: def __iter__() file: data_loader.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 186 unit: def camera_zero_bin() file: lib/actions.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 187 unit: def discretize_camera() file: lib/actions.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 188 unit: def undiscretize_camera() file: lib/actions.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 189 unit: def item_embed_id_to_name() file: lib/actions.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 190 unit: def preproc_qkv() file: lib/xf.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 4 id: 191 unit: def preproc_r() file: lib/xf.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 192 unit: def quick_gelu() file: lib/xf.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 193 unit: def empty_state() file: lib/xf.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 194 unit: def forward() file: lib/xf.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 195 unit: def from_factored() file: lib/action_mapping.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 196 unit: def to_factored() file: lib/action_mapping.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 197 unit: def get_action_space_update() file: lib/action_mapping.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 198 unit: def get_zero_action() file: lib/action_mapping.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 199 unit: def from_factored() file: lib/action_mapping.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 200 unit: def to_factored() file: lib/action_mapping.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 201 unit: def get_zero_action() file: lib/action_mapping.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 202 unit: def get_zero_action() file: lib/action_mapping.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 203 unit: def forward() file: lib/policy.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 204 unit: def output_latent_size() file: lib/policy.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 205 unit: def make_value_head() file: lib/policy.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 4 id: 206 unit: def make_action_head() file: lib/policy.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 207 unit: def initial_state() file: lib/policy.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 208 unit: def get_kl_of_action_dists() file: lib/policy.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 209 unit: def make_action_head() file: lib/policy.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 210 unit: def initial_state() file: lib/policy.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 211 unit: def __init__() file: lib/scaled_mse_head.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 0 id: 212 unit: def forward() file: lib/scaled_mse_head.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 213 unit: def loss() file: lib/scaled_mse_head.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 214 unit: def denormalize() file: lib/scaled_mse_head.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 215 unit: def normalize() file: lib/scaled_mse_head.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 216 unit: def have_cuda() file: lib/torch_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 0 id: 217 unit: def default_device_type() file: lib/torch_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 0 id: 218 unit: def dev() file: lib/torch_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 0 id: 219 unit: def zeros() file: lib/torch_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 220 unit: def ones() file: lib/torch_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 221 unit: def arange() file: lib/torch_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 222 unit: def forward() file: lib/torch_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 2 id: 223 unit: def forward() file: lib/torch_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 224 unit: def output_shape() file: lib/mlp.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 225 unit: def get_log_keys() file: lib/masked_attention.py start line: 0 end line: 0 size: 2 LOC McCabe index: 3 number of parameters: 1 id: 226 unit: def forward() file: lib/action_head.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 227 unit: def logprob() file: lib/action_head.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 228 unit: def entropy() file: lib/action_head.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 229 unit: def sample() file: lib/action_head.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 230 unit: def kl_divergence() file: lib/action_head.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 231 unit: def __init__() file: lib/action_head.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 0 id: 232 unit: def logprob() file: lib/action_head.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 3 id: 233 unit: def sample() file: lib/action_head.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 3 id: 234 unit: def entropy() file: lib/action_head.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 2 id: 235 unit: def kl_divergence() file: lib/action_head.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 3 id: 236 unit: def _num_leaves() file: lib/tree_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 1 id: 237 unit: def __hash__() file: lib/tree_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 238 unit: def __ne__() file: lib/tree_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 239 unit: def __repr__() file: lib/tree_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 240 unit: def __eq__() file: lib/tree_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 241 unit: def tuple_to_iterable() file: lib/tree_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 242 unit: def tuple_from_iterable() file: lib/tree_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 243 unit: def list_to_iterable() file: lib/tree_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 244 unit: def list_from_iterable() file: lib/tree_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 245 unit: def dict_from_iterable() file: lib/tree_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 246 unit: def ordered_dict_from_iterable() file: lib/tree_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 247 unit: def default_dict_to_iterable() file: lib/tree_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 248 unit: def default_dict_from_iterable() file: lib/tree_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 249 unit: def none_to_iterable() file: lib/tree_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 250 unit: def none_from_iterable() file: lib/tree_util.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 251 unit: def reset() file: inverse_dynamics_model.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1