SYMBOL INDEX (192 symbols across 11 files)

FILE: r2r_src/agent.py
  class BaseAgent (line 27) | class BaseAgent(object):
    method __init__ (line 30) | def __init__(self, env, results_path):
    method write_results (line 37) | def write_results(self):
    method get_results (line 42) | def get_results(self):
    method rollout (line 46) | def rollout(self, **args):
    method get_agent (line 51) | def get_agent(name):
    method test (line 54) | def test(self, iters=None, **kwargs):
  class Seq2SeqAgent (line 79) | class Seq2SeqAgent(BaseAgent):
    method __init__ (line 94) | def __init__(self, env, results_path, tok, episode_len=20):
    method _sort_batch (line 123) | def _sort_batch(self, obs):
    method _feature_variable (line 142) | def _feature_variable(self, obs):
    method _candidate_variable (line 149) | def _candidate_variable(self, obs):
    method get_input_feat (line 160) | def get_input_feat(self, obs):
    method _teacher_action (line 170) | def _teacher_action(self, obs, ended):
    method make_equiv_action (line 191) | def make_equiv_action(self, a_t, perm_obs, perm_idx=None, traj=None):
    method rollout (line 229) | def rollout(self, train_ml=None, train_rl=True, reset=True):
    method test (line 474) | def test(self, use_dropout=False, feedback='argmax', allow_cheat=False...
    method zero_grad (line 485) | def zero_grad(self):
    method accumulate_gradient (line 492) | def accumulate_gradient(self, feedback='teacher', **kwargs):
    method optim_step (line 504) | def optim_step(self):
    method train (line 512) | def train(self, n_iters, feedback='teacher', **kwargs):
    method save (line 549) | def save(self, epoch, path):
    method load (line 566) | def load(self, path):

FILE: r2r_src/env.py
  class EnvBatch (line 23) | class EnvBatch():
    method __init__ (line 27) | def __init__(self, feature_store=None, batch_size=100):
    method _make_id (line 58) | def _make_id(self, scanId, viewpointId):
    method newEpisodes (line 61) | def newEpisodes(self, scanIds, viewpointIds, headings):
    method getStates (line 65) | def getStates(self):
    method makeActions (line 84) | def makeActions(self, actions):
  class R2RBatch (line 91) | class R2RBatch():
    method __init__ (line 94) | def __init__(self, feature_store, batch_size=100, seed=10, splits=['tr...
    method size (line 162) | def size(self):
    method _load_nav_graphs (line 165) | def _load_nav_graphs(self):
    method _next_minibatch (line 183) | def _next_minibatch(self, tile_one=False, batch_size=None, **kwargs):
    method reset_epoch (line 207) | def reset_epoch(self, shuffle=False):
    method _shortest_path_action (line 214) | def _shortest_path_action(self, state, goalViewpointId):
    method make_candidate (line 222) | def make_candidate(self, feature, scanId, viewpointId, viewId):
    method _get_obs (line 294) | def _get_obs(self):
    method reset (line 329) | def reset(self, batch=None, inject=False, **kwargs):
    method step (line 345) | def step(self, actions):
    method get_statistics (line 350) | def get_statistics(self):

FILE: r2r_src/eval.py
  class Evaluation (line 17) | class Evaluation(object):
    method __init__ (line 20) | def __init__(self, splits, scans, tok):
    method _get_nearest (line 41) | def _get_nearest(self, scan, goal_id, path):
    method _score_item (line 51) | def _score_item(self, instr_id, path):
    method score (line 74) | def score(self, output_file):

FILE: r2r_src/model_OSCAR.py
  class VLNBERT (line 10) | class VLNBERT(nn.Module):
    method __init__ (line 11) | def __init__(self, feature_size=2048+128):
    method forward (line 28) | def forward(self, mode, sentence, token_type_ids=None,
  class BertLayerNorm (line 60) | class BertLayerNorm(nn.Module):
    method __init__ (line 61) | def __init__(self, hidden_size, eps=1e-12):
    method forward (line 69) | def forward(self, x):
  class Critic (line 76) | class Critic(nn.Module):
    method __init__ (line 77) | def __init__(self):
    method forward (line 86) | def forward(self, state):

FILE: r2r_src/model_PREVALENT.py
  class VLNBERT (line 12) | class VLNBERT(nn.Module):
    method __init__ (line 13) | def __init__(self, feature_size=2048+128):
    method forward (line 35) | def forward(self, mode, sentence, token_type_ids=None,
  class BertLayerNorm (line 69) | class BertLayerNorm(nn.Module):
    method __init__ (line 70) | def __init__(self, hidden_size, eps=1e-12):
    method forward (line 78) | def forward(self, x):
  class Critic (line 85) | class Critic(nn.Module):
    method __init__ (line 86) | def __init__(self):
    method forward (line 95) | def forward(self, state):

FILE: r2r_src/param.py
  class Param (line 5) | class Param:
    method __init__ (line 6) | def __init__(self):

FILE: r2r_src/train.py
  function train (line 41) | def train(train_env, tok, n_iters, log_every=2000, val_envs={}, aug_env=...
  function valid (line 151) | def valid(train_env, tok, val_envs={}):
  function setup (line 178) | def setup():
  function train_val (line 184) | def train_val(test_only=False):
  function train_val_augment (line 222) | def train_val_augment(test_only=False):

FILE: r2r_src/utils.py
  function load_nav_graphs (line 23) | def load_nav_graphs(scans):
  function load_datasets (line 51) | def load_datasets(splits):
  function pad_instr_tokens (line 91) | def pad_instr_tokens(instr_tokens, maxlength=20):
  class Tokenizer (line 108) | class Tokenizer(object):
    method __init__ (line 112) | def __init__(self, vocab=None, encoding_length=20):
    method finalize (line 132) | def finalize(self):
    method add_word (line 138) | def add_word(self, word):
    method split_sentence (line 144) | def split_sentence(sentence):
    method vocab_size (line 155) | def vocab_size(self):
    method encode_sentence (line 158) | def encode_sentence(self, sentence, max_length=None):
    method decode_sentence (line 180) | def decode_sentence(self, encoding, length=None):
    method shrink (line 191) | def shrink(self, inst):
  function build_vocab (line 208) | def build_vocab(splits=['train'], min_count=5, start_vocab=base_vocab):
  function write_vocab (line 225) | def write_vocab(vocab, path):
  function read_vocab (line 232) | def read_vocab(path):
  function asMinutes (line 238) | def asMinutes(s):
  function timeSince (line 244) | def timeSince(since, percent):
  function read_img_features (line 251) | def read_img_features(feature_store, test_only=False):
  function read_candidates (line 282) | def read_candidates(candidates_store):
  function add_exploration (line 322) | def add_exploration(paths):
  function angle_feature (line 329) | def angle_feature(heading, elevation):
  function new_simulator (line 339) | def new_simulator():
  function get_point_angle_feature (line 355) | def get_point_angle_feature(baseViewId=0):
  function get_all_point_angle_feature (line 376) | def get_all_point_angle_feature():
  function add_idx (line 379) | def add_idx(inst):
  class GracefulKiller (line 384) | class GracefulKiller:
    method __init__ (line 386) | def __init__(self):
    method exit_gracefully (line 390) | def exit_gracefully(self,signum, frame):
  class Timer (line 395) | class Timer:
    method __init__ (line 396) | def __init__(self):
    method reset (line 401) | def reset(self):
    method tic (line 406) | def tic(self, key):
    method toc (line 409) | def toc(self, key):
    method step (line 416) | def step(self):
    method show (line 419) | def show(self):
  function stop_words_location (line 432) | def stop_words_location(inst, mask=False):
  function get_segments (line 442) | def get_segments(inst, mask=False):
  function clever_pad_sequence (line 450) | def clever_pad_sequence(sequences, batch_first=True, padding_value=0):
  function length2mask (line 471) | def length2mask(length, size=None):
  function average_length (line 478) | def average_length(path2inst):
  function tile_batch (line 486) | def tile_batch(tensor, multiplier):
  function viewpoint_drop_mask (line 491) | def viewpoint_drop_mask(viewpoint, seed=None, drop_func=None):
  class FloydGraph (line 498) | class FloydGraph:
    method __init__ (line 499) | def __init__(self):
    method distance (line 504) | def distance(self, x, y):
    method add_edge (line 510) | def add_edge(self, x, y, dis):
    method update (line 517) | def update(self, k):
    method visited (line 528) | def visited(self, k):
    method path (line 531) | def path(self, x, y):
  function print_progress (line 549) | def print_progress(iteration, total, prefix='', suffix='', decimals=1, b...
  function ndtw_initialize (line 571) | def ndtw_initialize():
  function ndtw_graphload (line 586) | def ndtw_graphload(scan):
  class DTW (line 619) | class DTW(object):
    method __init__ (line 632) | def __init__(self, graph, weight='weight', threshold=3.0):
    method __call__ (line 645) | def __call__(self, prediction, reference, metric='sdtw'):

FILE: r2r_src/vlnbert/vlnbert_OSCAR.py
  class CaptionBertSelfAttention (line 20) | class CaptionBertSelfAttention(BertSelfAttention):
    method __init__ (line 24) | def __init__(self, config):
    method forward (line 28) | def forward(self, mode, hidden_states, attention_mask, head_mask=None,
  class CaptionBertAttention (line 75) | class CaptionBertAttention(BertAttention):
    method __init__ (line 79) | def __init__(self, config):
    method forward (line 85) | def forward(self, mode, input_tensor, attention_mask, head_mask=None,
  class CaptionBertLayer (line 101) | class CaptionBertLayer(BertLayer):
    method __init__ (line 105) | def __init__(self, config):
    method forward (line 111) | def forward(self, mode, hidden_states, attention_mask, head_mask=None,
  class CaptionBertEncoder (line 126) | class CaptionBertEncoder(BertEncoder):
    method __init__ (line 130) | def __init__(self, config):
    method forward (line 138) | def forward(self, mode, hidden_states, attention_mask, head_mask=None,
  class BertImgModel (line 176) | class BertImgModel(BertPreTrainedModel):
    method __init__ (line 179) | def __init__(self, config):
    method forward (line 190) | def forward(self, mode, input_ids, token_type_ids=None, attention_mask...
  class VLNBert (line 226) | class VLNBert(BertPreTrainedModel):
    method __init__ (line 230) | def __init__(self, config):
    method forward (line 242) | def forward(self, mode, input_ids, token_type_ids=None, attention_mask...

FILE: r2r_src/vlnbert/vlnbert_PREVALENT.py
  function gelu (line 22) | def gelu(x):
  function swish (line 31) | def swish(x):
  class BertEmbeddings (line 44) | class BertEmbeddings(nn.Module):
    method __init__ (line 47) | def __init__(self, config):
    method forward (line 58) | def forward(self, input_ids, token_type_ids=None, position_ids=None):
  class BertSelfAttention (line 76) | class BertSelfAttention(nn.Module):
    method __init__ (line 77) | def __init__(self, config):
    method transpose_for_scores (line 95) | def transpose_for_scores(self, x):
    method forward (line 100) | def forward(self, hidden_states, attention_mask, head_mask=None):
  class BertSelfOutput (line 136) | class BertSelfOutput(nn.Module):
    method __init__ (line 137) | def __init__(self, config):
    method forward (line 143) | def forward(self, hidden_states, input_tensor):
  class BertAttention (line 150) | class BertAttention(nn.Module):
    method __init__ (line 151) | def __init__(self, config):
    method forward (line 156) | def forward(self, input_tensor, attention_mask, head_mask=None):
  class BertIntermediate (line 163) | class BertIntermediate(nn.Module):
    method __init__ (line 164) | def __init__(self, config):
    method forward (line 172) | def forward(self, hidden_states):
  class BertOutput (line 178) | class BertOutput(nn.Module):
    method __init__ (line 179) | def __init__(self, config):
    method forward (line 185) | def forward(self, hidden_states, input_tensor):
  class BertLayer (line 192) | class BertLayer(nn.Module):
    method __init__ (line 193) | def __init__(self, config):
    method forward (line 199) | def forward(self, hidden_states, attention_mask, head_mask=None):
  class BertPooler (line 208) | class BertPooler(nn.Module):
    method __init__ (line 209) | def __init__(self, config):
    method forward (line 214) | def forward(self, hidden_states):
  class BertXAttention (line 223) | class BertXAttention(nn.Module):
    method __init__ (line 224) | def __init__(self, config, ctx_dim=None):
    method forward (line 229) | def forward(self, input_tensor, ctx_tensor, ctx_att_mask=None):
  class BertOutAttention (line 235) | class BertOutAttention(nn.Module):
    method __init__ (line 236) | def __init__(self, config, ctx_dim=None):
    method transpose_for_scores (line 255) | def transpose_for_scores(self, x):
    method forward (line 260) | def forward(self, hidden_states, context, attention_mask=None):
  class LXRTXLayer (line 291) | class LXRTXLayer(nn.Module):
    method __init__ (line 292) | def __init__(self, config):
    method cross_att (line 306) | def cross_att(self, lang_input, lang_attention_mask, visn_input, visn_...
    method self_att (line 311) | def self_att(self, visn_input, visn_attention_mask):
    method output_fc (line 316) | def output_fc(self, visn_input):
    method forward (line 322) | def forward(self, lang_feats, lang_attention_mask,
  class VisionEncoder (line 345) | class VisionEncoder(nn.Module):
    method __init__ (line 346) | def __init__(self, vision_size, config):
    method forward (line 356) | def forward(self, visn_input):
  class VLNBert (line 366) | class VLNBert(BertPreTrainedModel):
    method __init__ (line 367) | def __init__(self, config):
    method forward (line 384) | def forward(self, mode, input_ids, token_type_ids=None,

FILE: r2r_src/vlnbert/vlnbert_init.py
  function get_tokenizer (line 5) | def get_tokenizer(args):
  function get_vlnbert_models (line 15) | def get_vlnbert_models(args, config=None):