SYMBOL INDEX (122 symbols across 10 files) FILE: training/data.py function load_vocab (line 28) | def load_vocab(path): function invert_dict (line 40) | def invert_dict(d): function flat_to_hierarchical_actions (line 53) | def flat_to_hierarchical_actions(actions, controller_action_lim): function _dataset_to_tensor (line 90) | def _dataset_to_tensor(dset, mask=None, dtype=np.int64): function eqaCollateCnn (line 101) | def eqaCollateCnn(batch): function eqaCollateSeq2seq (line 116) | def eqaCollateSeq2seq(batch): class EqaDataset (line 133) | class EqaDataset(Dataset): method __init__ (line 134) | def __init__(self, method _pick_envs_to_load (line 269) | def _pick_envs_to_load(self, method _load_envs (line 286) | def _load_envs(self, start_idx=-1, in_order=False): method _clear_api_threads (line 365) | def _clear_api_threads(self): method _clear_memory (line 370) | def _clear_memory(self): method _check_if_all_envs_loaded (line 379) | def _check_if_all_envs_loaded(self): method set_camera (line 388) | def set_camera(self, e, pos, robot_height=1.0): method render (line 398) | def render(self, e): method get_frames (line 401) | def get_frames(self, e, pos_queue, preprocess=True): method get_hierarchical_features_till_spawn (line 418) | def get_hierarchical_features_till_spawn(self, actions, backtrack_step... method __getitem__ (line 459) | def __getitem__(self, index): method __len__ (line 798) | def __len__(self): class EqaDataLoader (line 805) | class EqaDataLoader(DataLoader): method __init__ (line 806) | def __init__(self, **kwargs): method close (line 889) | def close(self): method __enter__ (line 892) | def __enter__(self): method __exit__ (line 895) | def __exit__(self, exc_type, exc_value, traceback): FILE: training/metrics.py class Metric (line 16) | class Metric(): method __init__ (line 17) | def __init__(self, info={}, metric_names=[], log_json=None): method update (line 28) | def update(self, values): method get_stat_string (line 58) | def get_stat_string(self, mode=1): method dump_log (line 72) | def dump_log(self): class VqaMetric (line 86) | class VqaMetric(Metric): method __init__ (line 87) | def __init__(self, info={}, metric_names=[], log_json=None): method compute_ranks (line 90) | def compute_ranks(self, scores, labels): class NavMetric (line 101) | class NavMetric(Metric): method __init__ (line 102) | def __init__(self, info={}, metric_names=[], log_json=None): FILE: training/models.py function build_mlp (line 22) | def build_mlp(input_dim, function get_state (line 50) | def get_state(m): function repackage_hidden (line 59) | def repackage_hidden(h, batch_size): function ensure_shared_grads (line 68) | def ensure_shared_grads(model, shared_model): class MaskedNLLCriterion (line 76) | class MaskedNLLCriterion(nn.Module): method __init__ (line 77) | def __init__(self): method forward (line 80) | def forward(self, input, target, mask): class MultitaskCNNOutput (line 89) | class MultitaskCNNOutput(nn.Module): method __init__ (line 90) | def __init__( method forward (line 160) | def forward(self, x): class MultitaskCNN (line 202) | class MultitaskCNN(nn.Module): method __init__ (line 203) | def __init__( method forward (line 273) | def forward(self, x): class QuestionLstmEncoder (line 320) | class QuestionLstmEncoder(nn.Module): method __init__ (line 321) | def __init__(self, method init_weights (line 343) | def init_weights(self): method forward (line 347) | def forward(self, x): class VqaLstmModel (line 371) | class VqaLstmModel(nn.Module): method __init__ (line 372) | def __init__(self, method forward (line 401) | def forward(self, questions): class VqaLstmCnnAttentionModel (line 407) | class VqaLstmCnnAttentionModel(nn.Module): method __init__ (line 408) | def __init__(self, method forward (line 451) | def forward(self, images, questions): class NavCnnModel (line 487) | class NavCnnModel(nn.Module): method __init__ (line 488) | def __init__(self, method forward (line 531) | def forward(self, img_feats, questions=None): class NavRnnMult (line 551) | class NavRnnMult(nn.Module): method __init__ (line 552) | def __init__(self, method init_hidden (line 613) | def init_hidden(self, bsz): method forward (line 626) | def forward(self, method step_forward (line 670) | def step_forward(self, img_feats, question_feats, actions_in, hidden): class NavRnn (line 705) | class NavRnn(nn.Module): method __init__ (line 706) | def __init__(self, method init_hidden (line 767) | def init_hidden(self, bsz): method forward (line 780) | def forward(self, method step_forward (line 823) | def step_forward(self, img_feats, question_feats, actions_in, hidden): class NavCnnRnnMultModel (line 856) | class NavCnnRnnMultModel(nn.Module): method __init__ (line 857) | def __init__( method forward (line 910) | def forward(self, class NavCnnRnnModel (line 943) | class NavCnnRnnModel(nn.Module): method __init__ (line 944) | def __init__( method forward (line 997) | def forward(self, class NavPlannerControllerModel (line 1030) | class NavPlannerControllerModel(nn.Module): method __init__ (line 1031) | def __init__(self, method forward (line 1092) | def forward(self, method planner_step (line 1151) | def planner_step(self, questions, img_feats, actions_in, planner_hidden): method controller_step (line 1161) | def controller_step(self, img_feats, actions_in, hidden_in): FILE: training/train_eqa.py function eval (line 29) | def eval(rank, args, shared_nav_model, shared_ans_model): function train (line 404) | def train(rank, args, shared_nav_model, shared_ans_model): FILE: training/train_nav.py function _rebuild_tensor_v2 (line 28) | def _rebuild_tensor_v2(storage, storage_offset, size, stride, requires_g... function eval (line 37) | def eval(rank, args, shared_model): function train (line 729) | def train(rank, args, shared_model): FILE: training/train_vqa.py function eval (line 28) | def eval(rank, args, shared_model): function train (line 157) | def train(rank, args, shared_model): FILE: training/utils/preprocess_questions.py function tokenize (line 21) | def tokenize(seq, function buildVocab (line 41) | def buildVocab(sequences, function encode (line 77) | def encode(seqTokens, tokenToIdx, allowUnk=False): function decode (line 89) | def decode(seqIdx, idxToToken, delim=None, stopAtEnd=True): function preprocessImages (line 101) | def preprocessImages(obj, render_dir=False): function processActions (line 123) | def processActions(actions): FILE: training/utils/preprocess_questions_pkl.py function tokenize (line 17) | def tokenize(seq, function buildVocab (line 37) | def buildVocab(sequences, function encode (line 73) | def encode(seqTokens, tokenToIdx, allowUnk=False): function decode (line 85) | def decode(seqIdx, idxToToken, delim=None, stopAtEnd=True): function preprocessImages (line 97) | def preprocessImages(obj, render_dir=False): function processActions (line 119) | def processActions(actions): FILE: utils/house3d.py class House3DUtils (line 20) | class House3DUtils(): method __init__ (line 21) | def __init__( method calibrate_steps (line 83) | def calibrate_steps(self, reset=True): method step (line 118) | def step(self, action, step_reward=False): method get_dist_to_target (line 159) | def get_dist_to_target(self, pos): method is_inside_room (line 167) | def is_inside_room(self, pos, room): method build_graph (line 180) | def build_graph(self, save_path=None): method load_graph (line 266) | def load_graph(self, path): method compute_shortest_path (line 280) | def compute_shortest_path(self, source, target, graph=None): method fit_grid_path_to_suncg (line 302) | def fit_grid_path_to_suncg(self, nodes, init_yaw=None, back_skip=2): method get_rotate_steps (line 395) | def get_rotate_steps(self, pos, target_yaw): method _vec_to_array (line 419) | def _vec_to_array(self, pos, yaw): method render_images_from_pos_queue (line 423) | def render_images_from_pos_queue(self, method render_video_from_pos_queue (line 478) | def render_video_from_pos_queue(self, method _parse (line 514) | def _parse(self, levelsToExplore=[0]): method spawn_room (line 592) | def spawn_room(self, room=None): method spawn_object (line 620) | def spawn_object(self, obj=None, room=None): method set_target_object (line 684) | def set_target_object(self, obj, room): method _load_semantic_classes (line 817) | def _load_semantic_classes(self, color_file=None): method _get_best_yaw_obj_from_pos (line 832) | def _get_best_yaw_obj_from_pos(self, obj_id, grid_pos, height=1.0): method _get_best_view_obj (line 863) | def _get_best_view_obj(self, FILE: utils/make_houses.py function extract_threaded (line 32) | def extract_threaded(house):