SYMBOL INDEX (584 symbols across 71 files)

FILE: arguments.py
  function get_args (line 6) | def get_args():

FILE: collect_data.py
  function collectData (line 13) | def collectData(device, train_data, config):

FILE: crowd_nav/configs/config.py
  class BaseConfig (line 4) | class BaseConfig(object):
    method __init__ (line 5) | def __init__(self):
  class Config (line 9) | class Config(object):

FILE: crowd_nav/policy/orca.py
  class ORCA (line 7) | class ORCA(Policy):
    method __init__ (line 8) | def __init__(self, config):
    method predict (line 64) | def predict(self, state):

FILE: crowd_nav/policy/policy.py
  class Policy (line 5) | class Policy(object):
    method __init__ (line 6) | def __init__(self, config):
    method predict (line 22) | def predict(self, state):
    method reach_destination (line 30) | def reach_destination(state):

FILE: crowd_nav/policy/policy_factory.py
  function none_policy (line 2) | def none_policy():

FILE: crowd_nav/policy/social_force.py
  class SOCIAL_FORCE (line 5) | class SOCIAL_FORCE(Policy):
    method __init__ (line 6) | def __init__(self, config):
    method predict (line 11) | def predict(self, state):

FILE: crowd_nav/policy/srnn.py
  class SRNN (line 7) | class SRNN(Policy):
    method __init__ (line 8) | def __init__(self, config):
    method clip_action (line 17) | def clip_action(self, raw_action, v_pref):
  class selfAttn_merge_SRNN (line 47) | class selfAttn_merge_SRNN(SRNN):
    method __init__ (line 48) | def __init__(self, config):

FILE: crowd_sim/envs/crowd_sim.py
  class CrowdSim (line 19) | class CrowdSim(gym.Env):
    method __init__ (line 24) | def __init__(self):
    method configure (line 88) | def configure(self, config):
    method set_robot (line 188) | def set_robot(self, robot):
    method generate_random_human_position (line 192) | def generate_random_human_position(self, human_num):
    method generate_circle_crossing_human (line 203) | def generate_circle_crossing_human(self):
    method update_last_human_states (line 243) | def update_last_human_states(self, human_visibility, reset):
    method get_true_human_states (line 276) | def get_true_human_states(self):
    method generate_robot_humans (line 287) | def generate_robot_humans(self, phase, human_num=None):
    method smooth_action (line 315) | def smooth_action(self, action):
    method reset (line 361) | def reset(self, phase='train', test_case=None):
    method update_human_goals_randomly (line 415) | def update_human_goals_randomly(self):
    method update_human_goal (line 453) | def update_human_goal(self, human):
    method calc_offset_angle (line 488) | def calc_offset_angle(self, state1, state2):
    method detect_visible (line 513) | def detect_visible(self, state1, state2, robot1 = False, custom_fov=No...
    method get_num_human_in_fov (line 558) | def get_num_human_in_fov(self):
    method last_human_states_obj (line 576) | def last_human_states_obj(self):
    method calc_reward (line 588) | def calc_reward(self, action):
    method generate_ob (line 663) | def generate_ob(self, reset):
    method get_human_actions (line 680) | def get_human_actions(self):
    method step (line 705) | def step(self, action, update=True):
    method render (line 750) | def render(self, mode='human'):

FILE: crowd_sim/envs/crowd_sim_pred.py
  class CrowdSimPred (line 11) | class CrowdSimPred(CrowdSimVarNum):
    method __init__ (line 16) | def __init__(self):
    method configure (line 27) | def configure(self, config):
    method reset (line 33) | def reset(self, phase='train', test_case=None):
    method set_robot (line 38) | def set_robot(self, robot):
    method generate_ob (line 62) | def generate_ob(self, reset, sort=True):
    method step (line 100) | def step(self, action, update=True):
    method calc_reward (line 216) | def calc_reward(self, action, danger_zone='future'):
    method render (line 236) | def render(self, mode='human'):

FILE: crowd_sim/envs/crowd_sim_pred_real_gst.py
  class CrowdSimPredRealGST (line 7) | class CrowdSimPredRealGST(CrowdSimPred):
    method __init__ (line 13) | def __init__(self):
    method set_robot (line 27) | def set_robot(self, robot):
    method talk2Env (line 64) | def talk2Env(self, data):
    method generate_ob (line 76) | def generate_ob(self, reset, sort=False):
    method calc_reward (line 97) | def calc_reward(self, action, danger_zone='future'):
    method render (line 104) | def render(self, mode='human'):

FILE: crowd_sim/envs/crowd_sim_var_num.py
  class CrowdSimVarNum (line 13) | class CrowdSimVarNum(CrowdSim):
    method __init__ (line 18) | def __init__(self):
    method configure (line 31) | def configure(self, config):
    method set_robot (line 37) | def set_robot(self, robot):
    method generate_robot_humans (line 64) | def generate_robot_humans(self, phase, human_num=None):
    method generate_circle_crossing_human (line 116) | def generate_circle_crossing_human(self):
    method calc_human_future_traj (line 152) | def calc_human_future_traj(self, method):
    method generate_ob (line 233) | def generate_ob(self, reset, sort=False):
    method update_human_pos_goal (line 284) | def update_human_pos_goal(self, human):
    method reset (line 303) | def reset(self, phase='train', test_case=None):
    method step (line 366) | def step(self, action, update=True):
    method calc_reward (line 465) | def calc_reward(self, action, danger_zone='circle'):
    method render (line 564) | def render(self, mode='human'):

FILE: crowd_sim/envs/crowd_sim_var_num_collect.py
  class CrowdSimVarNumCollect (line 11) | class CrowdSimVarNumCollect(CrowdSimVarNum):
    method __init__ (line 17) | def __init__(self):
    method set_robot (line 27) | def set_robot(self, robot):
    method reset (line 43) | def reset(self, phase='train', test_case=None):
    method generate_ob (line 98) | def generate_ob(self, reset, sort=False):
    method calc_reward (line 136) | def calc_reward(self, action, danger_zone='circle'):

FILE: crowd_sim/envs/ros_turtlebot2i_env.py
  class rosTurtlebot2iEnv (line 27) | class rosTurtlebot2iEnv(CrowdSimPredRealGST):
    method __init__ (line 34) | def __init__(self):
    method configure (line 66) | def configure(self, config):
    method set_robot (line 103) | def set_robot(self, robot):
    method set_ob_act_space (line 106) | def set_ob_act_space(self):
    method state_cb (line 147) | def state_cb(self, jointStateMsg, humanArrayMsg):
    method state_cb_dummy (line 154) | def state_cb_dummy(self, jointStateMsg):
    method readMsg (line 157) | def readMsg(self):
    method reset (line 248) | def reset(self):
    method smooth (line 292) | def smooth(self, v, w):
    method generate_ob (line 312) | def generate_ob(self):
    method step (line 339) | def step(self, action, update=True):
    method render (line 430) | def render(self, mode='human'):
    method shutdown (line 520) | def shutdown(self):
    method smoothStop (line 526) | def smoothStop(self):

FILE: crowd_sim/envs/utils/agent.py
  class Agent (line 10) | class Agent(object):
    method __init__ (line 11) | def __init__(self, config, section):
    method print_info (line 39) | def print_info(self):
    method sample_random_attributes (line 44) | def sample_random_attributes(self):
    method set (line 52) | def set(self, px, py, gx, gy, vx, vy, theta, radius=None, v_pref=None):
    method set_list (line 67) | def set_list(self, px, py, vx, vy, radius, gx, gy, v_pref, theta):
    method get_observable_state (line 78) | def get_observable_state(self):
    method get_observable_state_list (line 81) | def get_observable_state_list(self):
    method get_observable_state_list_noV (line 84) | def get_observable_state_list_noV(self):
    method get_next_observable_state (line 87) | def get_next_observable_state(self, action):
    method get_full_state (line 100) | def get_full_state(self):
    method get_full_state_list (line 103) | def get_full_state_list(self):
    method get_full_state_list_noV (line 106) | def get_full_state_list_noV(self):
    method get_position (line 110) | def get_position(self):
    method set_position (line 113) | def set_position(self, position):
    method get_goal_position (line 117) | def get_goal_position(self):
    method get_velocity (line 120) | def get_velocity(self):
    method set_velocity (line 124) | def set_velocity(self, velocity):
    method act (line 130) | def act(self, ob):
    method check_validity (line 137) | def check_validity(self, action):
    method compute_position (line 143) | def compute_position(self, action, delta_t):
    method step (line 170) | def step(self, action):
    method one_step_lookahead (line 185) | def one_step_lookahead(self, pos, action):
    method reached_destination (line 194) | def reached_destination(self):

FILE: crowd_sim/envs/utils/human.py
  class Human (line 5) | class Human(Agent):
    method __init__ (line 7) | def __init__(self, config, section):
    method act (line 14) | def act(self, ob):
    method act_joint_state (line 26) | def act_joint_state(self, ob):

FILE: crowd_sim/envs/utils/info.py
  class Timeout (line 1) | class Timeout(object):
    method __init__ (line 2) | def __init__(self):
    method __str__ (line 5) | def __str__(self):
  class ReachGoal (line 9) | class ReachGoal(object):
    method __init__ (line 10) | def __init__(self):
    method __str__ (line 13) | def __str__(self):
  class Danger (line 17) | class Danger(object):
    method __init__ (line 18) | def __init__(self, min_dist):
    method __str__ (line 21) | def __str__(self):
  class Collision (line 25) | class Collision(object):
    method __init__ (line 26) | def __init__(self):
    method __str__ (line 29) | def __str__(self):
  class OutRoad (line 32) | class OutRoad(object):
    method __init__ (line 33) | def __init__(self):
    method __str__ (line 36) | def __str__(self):
  class Nothing (line 39) | class Nothing(object):
    method __init__ (line 40) | def __init__(self):
    method __str__ (line 43) | def __str__(self):

FILE: crowd_sim/envs/utils/recorder.py
  class Recoder (line 5) | class Recoder(object):
    method __init__ (line 6) | def __init__(self):
    method saveEpisode (line 22) | def saveEpisode(self,episodeCounter):
    method loadActions (line 52) | def loadActions(self):
    method clear (line 58) | def clear(self):
  class humanRecoder (line 66) | class humanRecoder(object):
    method __init__ (line 67) | def __init__(self, human_num):
    method addInitPos (line 78) | def addInitPos(self, human_id, px, py, radius, v_pref):
    method addGoalPos (line 82) | def addGoalPos(self, human_id, px, py):
    method loadLists (line 85) | def loadLists(self, initPos, goalPos):
    method getInitList (line 89) | def getInitList(self):
    method getGoalList (line 92) | def getGoalList(self):
    method getInitPos (line 95) | def getInitPos(self, human_id):
    method getNextGoalPos (line 100) | def getNextGoalPos(self, human_id):
    method goalIsEmpty (line 107) | def goalIsEmpty(self, human_id):
    method clear (line 113) | def clear(self):
  class jointStateRecoder (line 117) | class jointStateRecoder(object):
    method __init__ (line 118) | def __init__(self, human_num):
    method add_traj (line 141) | def add_traj(self, seed, ob, srnn):
    method add_label (line 168) | def add_label(self, seed, respond_traj_len, traj_len, respond_time, to...
    method save_to_file (line 176) | def save_to_file(self, directory):
    method clear (line 204) | def clear(self):

FILE: crowd_sim/envs/utils/robot.py
  class Robot (line 5) | class Robot(Agent):
    method __init__ (line 6) | def __init__(self, config,section):
    method act (line 10) | def act(self, ob):
    method actWithJointState (line 19) | def actWithJointState(self,ob):

FILE: crowd_sim/envs/utils/state.py
  class JointState (line 10) | class JointState(object):
    method __init__ (line 13) | def __init__(self, self_state, human_states):
    method to_flatten_list (line 32) | def to_flatten_list(self):

FILE: gst_updated/scripts/experiments/eval.py
  function main (line 12) | def main(args):
  function inference (line 56) | def inference(loader, model, args, mode='val', tau=1., device='cuda:0'):

FILE: gst_updated/scripts/experiments/eval_trajnet.py
  function main (line 12) | def main(args):
  function test (line 56) | def test(loader, model, tau=0.03, device='cuda:0'):
  function inference (line 85) | def inference(loader, model, args, mode='val', tau=1., device='cuda:0'):

FILE: gst_updated/scripts/experiments/test.py
  function main (line 12) | def main(args):
  function inference (line 49) | def inference(loader, model, args, mode='val', tau=1., device='cuda:0'):

FILE: gst_updated/scripts/experiments/test_gst.py
  function main (line 12) | def main(args):
  function inference (line 49) | def inference(loader, model, args, mode='val', tau=1., device='cuda:0'):

FILE: gst_updated/scripts/experiments/train.py
  function main (line 17) | def main(args):
  function train (line 50) | def train(args, data_loaders, writer, logdir, device='cuda:0'):

FILE: gst_updated/scripts/wrapper/crowd_nav_interface_multi_env_parallel.py
  class CrowdNavPredInterfaceMultiEnv (line 11) | class CrowdNavPredInterfaceMultiEnv(object):
    method __init__ (line 13) | def __init__(self, load_path, device, config, num_env):
    method forward (line 35) | def forward(self, input_traj,input_binary_mask, sampling = True):

FILE: gst_updated/scripts/wrapper/crowd_nav_interface_multi_env_visualization_test_single_batch.py
  function load_batch_dataset (line 15) | def load_batch_dataset(pkg_path, subfolder='test', num_workers=4, shuffl...
  function visualize_trajectory (line 101) | def visualize_trajectory(obs_traj, loss_mask, sample_index, obs_seq_len=...
  function CrowdNavPredInterfaceMultiEnv (line 130) | def CrowdNavPredInterfaceMultiEnv(
  function visualize_output_trajectory_deterministic (line 331) | def visualize_output_trajectory_deterministic(input_traj, input_binary_m...

FILE: gst_updated/scripts/wrapper/crowd_nav_interface_parallel.py
  function seq_to_graph (line 9) | def seq_to_graph(seq_, seq_rel, attn_mech='rel_conv'):
  class CrowdNavPredInterfaceMultiEnv (line 23) | class CrowdNavPredInterfaceMultiEnv(object):
    method __init__ (line 25) | def __init__(self, load_path, device, config, num_env):
    method forward (line 45) | def forward(self, input_traj,input_binary_mask, sampling = True):
  function visualize_output_trajectory_deterministic (line 117) | def visualize_output_trajectory_deterministic(input_traj, input_binary_m...

FILE: gst_updated/src/gumbel_social_transformer/edge_selector_ghost.py
  class EdgeSelector (line 8) | class EdgeSelector(nn.Module):
    method __init__ (line 10) | def __init__(self, d_motion, d_model, nhead=4, dropout=0.1, activation...
    method forward (line 27) | def forward(self, x, A, attn_mask, tau=1., hard=False, device='cuda:0'):
    method edge_sampler (line 101) | def edge_sampler(self, edge_multinomial, tau=1., hard=False):

FILE: gst_updated/src/gumbel_social_transformer/edge_selector_no_ghost.py
  class EdgeSelector (line 8) | class EdgeSelector(nn.Module):
    method __init__ (line 10) | def __init__(self, d_motion, d_model, nhead=4, dropout=0.1, activation...
    method forward (line 26) | def forward(self, x, A, attn_mask, tau=1., hard=False, device='cuda:0'):
    method edge_sampler (line 91) | def edge_sampler(self, edge_multinomial, tau=1., hard=False):

FILE: gst_updated/src/gumbel_social_transformer/gumbel_social_transformer.py
  class GumbelSocialTransformer (line 5) | class GumbelSocialTransformer(nn.Module):
    method __init__ (line 6) | def __init__(self, d_motion, d_model, nhead_nodes, nhead_edges, nlayer...
    method forward (line 43) | def forward(self, x, A, attn_mask, tau=1., hard=False, device='cuda:0'):

FILE: gst_updated/src/gumbel_social_transformer/mha.py
  function multi_head_attention_forward (line 12) | def multi_head_attention_forward(
  class VanillaMultiheadAttention (line 259) | class VanillaMultiheadAttention(Module):
    method __init__ (line 263) | def __init__(self, embed_dim, num_heads, dropout=0., bias=True, add_bi...
    method _reset_parameters (line 303) | def _reset_parameters(self):
    method __setstate__ (line 319) | def __setstate__(self, state):
    method forward (line 325) | def forward(self, query: Tensor, key: Tensor, value: Tensor, key_paddi...

FILE: gst_updated/src/gumbel_social_transformer/node_encoder_layer_ghost.py
  class NodeEncoderLayer (line 6) | class NodeEncoderLayer(nn.Module):
    method __init__ (line 8) | def __init__(self, d_model, nhead, dim_feedforward=512, dropout=0.1, a...
    method forward (line 26) | def forward(self, x, sampled_edges, attn_mask, device="cuda:0"):

FILE: gst_updated/src/gumbel_social_transformer/node_encoder_layer_no_ghost.py
  class NodeEncoderLayer (line 6) | class NodeEncoderLayer(nn.Module):
    method __init__ (line 8) | def __init__(self, d_model, nhead, dim_feedforward=512, dropout=0.1, a...
    method forward (line 25) | def forward(self, x, sampled_edges, attn_mask, device="cuda:0"):

FILE: gst_updated/src/gumbel_social_transformer/st_model.py
  function offset_error_square_full_partial (line 15) | def offset_error_square_full_partial(x_pred, x_target, loss_mask_ped, lo...
  function negative_log_likelihood_full_partial (line 62) | def negative_log_likelihood_full_partial(gaussian_params, x_target, loss...
  class st_model (line 115) | class st_model(nn.Module):
    method __init__ (line 117) | def __init__(self, args, device='cuda:0'):
    method raw2gaussian (line 192) | def raw2gaussian(self, prob_raw):
    method sample_gaussian (line 211) | def sample_gaussian(self, gaussian_params, device='cuda:0', detach_sam...
    method edge_evolution (line 246) | def edge_evolution(self, xt_plus, At, device='cuda:0'):
    method forward (line 271) | def forward(self, x, A, attn_mask, loss_mask_rel, tau=1., hard=False, ...

FILE: gst_updated/src/gumbel_social_transformer/temperature_scheduler.py
  class Temp_Scheduler (line 1) | class Temp_Scheduler(object):
    method __init__ (line 2) | def __init__(self, total_epochs, curr_temp, base_temp, temp_min=0.33, ...
    method step (line 10) | def step(self, epoch=None):
    method decay_whole_process (line 13) | def decay_whole_process(self, epoch=None):

FILE: gst_updated/src/gumbel_social_transformer/temporal_convolution_net.py
  class TemporalConvolutionNet (line 4) | class TemporalConvolutionNet(nn.Module):
    method __init__ (line 5) | def __init__(
    method forward (line 40) | def forward(self, x):

FILE: gst_updated/src/gumbel_social_transformer/utils.py
  function _get_activation_fn (line 8) | def _get_activation_fn(activation):
  function _get_clones (line 15) | def _get_clones(module, N):
  function gumbel_softmax (line 18) | def gumbel_softmax(logits, tau=1, hard=False, eps=1e-10):
  function gumbel_softmax_sample (line 32) | def gumbel_softmax_sample(logits, tau=1, eps=1e-10):
  function sample_gumbel (line 39) | def sample_gumbel(shape, eps=1e-10):

FILE: gst_updated/src/mgnn/batch_trajectories.py
  class BatchTrajectoriesDataset (line 3) | class BatchTrajectoriesDataset(Dataset):
    method __init__ (line 4) | def __init__(
    method add_batch (line 22) | def add_batch(
    method __len__ (line 51) | def __len__(self):
    method __getitem__ (line 55) | def __getitem__(self, index):

FILE: gst_updated/src/mgnn/trajectories.py
  class TrajectoriesDataset (line 8) | class TrajectoriesDataset(Dataset):
    method __init__ (line 9) | def __init__(
    method __len__ (line 145) | def __len__(self):
    method __getitem__ (line 149) | def __getitem__(self, index):
  function read_file (line 163) | def read_file(_path, delim='\t'):

FILE: gst_updated/src/mgnn/trajectories_sdd.py
  class TrajectoriesDataset (line 8) | class TrajectoriesDataset(Dataset):
    method __init__ (line 9) | def __init__(
    method __len__ (line 146) | def __len__(self):
    method __getitem__ (line 150) | def __getitem__(self, index):
  function read_file (line 164) | def read_file(_path, delim='\t'):
  function read_sdd_file (line 177) | def read_sdd_file(_path):

FILE: gst_updated/src/mgnn/trajectories_trajnet.py
  class TrajectoriesDataset (line 9) | class TrajectoriesDataset(Dataset):
    method __init__ (line 10) | def __init__(
    method __len__ (line 177) | def __len__(self):
    method __getitem__ (line 181) | def __getitem__(self, index):
  function read_file (line 195) | def read_file(_path, delim='\t'):
  function read_trajnet_file (line 209) | def read_trajnet_file(_path):

FILE: gst_updated/src/mgnn/trajectories_trajnet_testset.py
  class TrajectoriesDataset (line 9) | class TrajectoriesDataset(Dataset):
    method __init__ (line 10) | def __init__(
    method __len__ (line 175) | def __len__(self):
    method __getitem__ (line 179) | def __getitem__(self, index):
  function read_file (line 203) | def read_file(_path, delim='\t'):
  function read_trajnet_file (line 217) | def read_trajnet_file(_path):

FILE: gst_updated/src/mgnn/utils.py
  function average_offset_error (line 8) | def average_offset_error(x_pred, x_target, loss_mask=None):
  function final_offset_error (line 18) | def final_offset_error(x_pred, x_target, loss_mask=None):
  function negative_log_likelihood (line 29) | def negative_log_likelihood(gaussian_params, x_target, loss_mask=None):
  function seq_to_graph (line 43) | def seq_to_graph(seq_, seq_rel, attn_mech='glob_kip'):
  function rotate_graph (line 75) | def rotate_graph(vtx, adj, theta):
  function random_rotate_graph (line 86) | def random_rotate_graph(args, vtx_obs, adj_obs, vtx_pred_gt, adj_pred_gt):
  function load_batch_dataset (line 100) | def load_batch_dataset(args, pkg_path, subfolder='train', num_workers=4,...
  function args2writername (line 127) | def args2writername(args):
  function arg_parse (line 154) | def arg_parse():

FILE: gst_updated/src/pec_net/sdd_trajectories.py
  class SDDTrajectoriesDataset (line 4) | class SDDTrajectoriesDataset(Dataset):
    method __init__ (line 5) | def __init__(
    method add_batch (line 23) | def add_batch(
    method __len__ (line 52) | def __len__(self):
    method __getitem__ (line 55) | def __getitem__(self, index):

FILE: gst_updated/src/pec_net/social_utils.py
  function naive_social (line 13) | def naive_social(p1_key, p2_key, all_data_dict):
  function find_min_time (line 19) | def find_min_time(t1, t2):
  function find_min_dist (line 34) | def find_min_dist(p1x, p1y, p2x, p2y):
  function social_and_temporal_filter (line 47) | def social_and_temporal_filter(p1_key, p2_key, all_data_dict, time_thres...
  function mark_similar (line 60) | def mark_similar(mask, sim_list):
  function collect_data (line 66) | def collect_data(set_name, dataset_type = 'image', batch_size=512, time_...
  function generate_pooled_data (line 137) | def generate_pooled_data(b_size, t_tresh, d_tresh, train=True, scene=Non...
  function initial_pos (line 152) | def initial_pos(traj_batches):
  function calculate_loss (line 160) | def calculate_loss(x, reconstructed_x, mean, log_var, criterion, future,...
  class SocialDataset (line 171) | class SocialDataset(data.Dataset):
    method __init__ (line 173) | def __init__(self, set_name, pkg_path, b_size=4096, t_tresh=60, d_tres...
  function split_square_block_matrix (line 227) | def split_square_block_matrix(block_mat):

FILE: rl/evaluation.py
  function evaluate (line 7) | def evaluate(actor_critic, eval_envs, num_processes, device, test_size, ...

FILE: rl/networks/distributions.py
  class FixedCategorical (line 18) | class FixedCategorical(torch.distributions.Categorical):
    method sample (line 19) | def sample(self):
    method log_probs (line 22) | def log_probs(self, actions):
    method mode (line 31) | def mode(self):
  class FixedNormal (line 36) | class FixedNormal(torch.distributions.Normal):
    method log_probs (line 37) | def log_probs(self, actions):
    method entrop (line 40) | def entrop(self):
    method mode (line 43) | def mode(self):
  class FixedBernoulli (line 48) | class FixedBernoulli(torch.distributions.Bernoulli):
    method log_probs (line 49) | def log_probs(self, actions):
    method entropy (line 52) | def entropy(self):
    method mode (line 55) | def mode(self):
  class Categorical (line 59) | class Categorical(nn.Module):
    method __init__ (line 60) | def __init__(self, num_inputs, num_outputs):
    method forward (line 71) | def forward(self, x):
  class DiagGaussian (line 76) | class DiagGaussian(nn.Module):
    method __init__ (line 77) | def __init__(self, num_inputs, num_outputs):
    method forward (line 86) | def forward(self, x):
  class Bernoulli (line 98) | class Bernoulli(nn.Module):
    method __init__ (line 99) | def __init__(self, num_inputs, num_outputs):
    method forward (line 107) | def forward(self, x):

FILE: rl/networks/dummy_vec_env.py
  class DummyVecEnv (line 5) | class DummyVecEnv(VecEnv):
    method __init__ (line 12) | def __init__(self, env_fns):
    method step_async (line 31) | def step_async(self, actions):
    method step_wait (line 45) | def step_wait(self):
    method reset (line 58) | def reset(self):
    method talk2Env_async (line 64) | def talk2Env_async(self, data):
    method talk2Env_wait (line 68) | def talk2Env_wait(self):
    method _save_obs (line 71) | def _save_obs(self, e, obs):
    method _obs_from_buf (line 78) | def _obs_from_buf(self):
    method get_images (line 81) | def get_images(self):
    method render (line 84) | def render(self, mode='human'):

FILE: rl/networks/envs.py
  function make_env (line 36) | def make_env(env_id, seed, rank, log_dir, allow_early_resets, config=Non...
  function make_vec_envs (line 97) | def make_vec_envs(env_name,
  class TimeLimitMask (line 144) | class TimeLimitMask(gym.Wrapper):
    method step (line 145) | def step(self, action):
    method reset (line 152) | def reset(self, **kwargs):
  class MaskGoal (line 157) | class MaskGoal(gym.ObservationWrapper):
    method observation (line 158) | def observation(self, observation):
  class TransposeObs (line 164) | class TransposeObs(gym.ObservationWrapper):
    method __init__ (line 165) | def __init__(self, env=None):
  class TransposeImage (line 172) | class TransposeImage(TransposeObs):
    method __init__ (line 173) | def __init__(self, env=None, op=[2, 0, 1]):
    method observation (line 189) | def observation(self, ob):
  class VecPyTorch (line 193) | class VecPyTorch(VecEnvWrapper):
    method __init__ (line 194) | def __init__(self, venv, device):
    method reset (line 200) | def reset(self):
    method step_async (line 209) | def step_async(self, actions):
    method step_wait (line 216) | def step_wait(self):
    method render_traj (line 226) | def render_traj(self, path, episode_num):
  class VecNormalize (line 234) | class VecNormalize(VecNormalize_):
    method __init__ (line 235) | def __init__(self, *args, **kwargs):
    method _obfilt (line 239) | def _obfilt(self, obs, update=True):
    method train (line 250) | def train(self):
    method eval (line 253) | def eval(self):
  class VecPyTorchFrameStack (line 259) | class VecPyTorchFrameStack(VecEnvWrapper):
    method __init__ (line 260) | def __init__(self, venv, nstack, device=None):
    method step_wait (line 279) | def step_wait(self):
    method reset (line 289) | def reset(self):
    method close (line 298) | def close(self):

FILE: rl/networks/model.py
  class Flatten (line 9) | class Flatten(nn.Module):
    method forward (line 10) | def forward(self, x):
  class Policy (line 14) | class Policy(nn.Module):
    method __init__ (line 16) | def __init__(self, obs_shape, action_space, base=None, base_kwargs=None):
    method is_recurrent (line 45) | def is_recurrent(self):
    method recurrent_hidden_state_size (line 49) | def recurrent_hidden_state_size(self):
    method forward (line 53) | def forward(self, inputs, rnn_hxs, masks):
    method act (line 56) | def act(self, inputs, rnn_hxs, masks, deterministic=False):
    method get_value (line 76) | def get_value(self, inputs, rnn_hxs, masks):
    method evaluate_actions (line 82) | def evaluate_actions(self, inputs, rnn_hxs, masks, action):

FILE: rl/networks/network_utils.py
  function get_render_func (line 10) | def get_render_func(venv):
  function get_vec_normalize (line 21) | def get_vec_normalize(venv):
  class AddBias (line 31) | class AddBias(nn.Module):
    method __init__ (line 32) | def __init__(self, bias):
    method forward (line 36) | def forward(self, x):
  function update_linear_schedule (line 45) | def update_linear_schedule(optimizer, epoch, total_num_epochs, initial_lr):
  function init (line 52) | def init(module, weight_init, bias_init, gain=1):
  function cleanup_log_dir (line 58) | def cleanup_log_dir(log_dir):

FILE: rl/networks/selfAttn_srnn_temp_node.py
  class SpatialEdgeSelfAttn (line 5) | class SpatialEdgeSelfAttn(nn.Module):
    method __init__ (line 10) | def __init__(self, args):
    method create_attn_mask (line 49) | def create_attn_mask(self, each_seq_len, seq_len, nenv, max_human_num):
    method forward (line 63) | def forward(self, inp, each_seq_len):
  class EdgeAttention_M (line 95) | class EdgeAttention_M(nn.Module):
    method __init__ (line 99) | def __init__(self, args):
    method create_attn_mask (line 132) | def create_attn_mask(self, each_seq_len, seq_len, nenv, max_human_num):
    method att_func (line 145) | def att_func(self, temporal_embed, spatial_embed, h_spatials, attn_mas...
    method forward (line 184) | def forward(self, h_temporal, h_spatials, each_seq_len):
  class EndRNN (line 225) | class EndRNN(RNNBase):
    method __init__ (line 229) | def __init__(self, args):
    method forward (line 262) | def forward(self, robot_s, h_spatial_other, h, masks):
  class selfAttn_merge_SRNN (line 287) | class selfAttn_merge_SRNN(nn.Module):
    method __init__ (line 291) | def __init__(self, obs_space_dict, args, infer=False):
    method forward (line 360) | def forward(self, inputs, rnn_hxs, masks, infer=False):
  function reshapeT (line 452) | def reshapeT(T, seq_length, nenv):

FILE: rl/networks/shmem_vec_env.py
  class ShmemVecEnv (line 21) | class ShmemVecEnv(VecEnv):
    method __init__ (line 26) | def __init__(self, env_fns, spaces=None, context='spawn'):
    method reset (line 62) | def reset(self):
    method step_async (line 70) | def step_async(self, actions):
    method step_wait (line 76) | def step_wait(self):
    method talk2Env_async (line 82) | def talk2Env_async(self, data):
    method talk2Env_wait (line 88) | def talk2Env_wait(self):
    method close_extras (line 93) | def close_extras(self):
    method get_images (line 104) | def get_images(self, mode='human'):
    method _decode_obses (line 109) | def _decode_obses(self, obs):
  function _subproc_worker (line 119) | def _subproc_worker(pipe, parent_pipe, env_fn_wrapper, obs_bufs, obs_sha...

FILE: rl/networks/srnn_model.py
  class RNNBase (line 10) | class RNNBase(nn.Module):
    method __init__ (line 15) | def __init__(self, args, edge):
    method _forward_gru (line 35) | def _forward_gru(self, x, hxs, masks):
  class HumanNodeRNN (line 108) | class HumanNodeRNN(RNNBase):
    method __init__ (line 112) | def __init__(self, args):
    method forward (line 149) | def forward(self, pos, h_temporal, h_spatial_other, h, masks):
  class HumanHumanEdgeRNN (line 177) | class HumanHumanEdgeRNN(RNNBase):
    method __init__ (line 181) | def __init__(self, args):
    method forward (line 202) | def forward(self, inp, h, masks):
  class EdgeAttention (line 219) | class EdgeAttention(nn.Module):
    method __init__ (line 223) | def __init__(self, args):
    method att_func (line 256) | def att_func(self, temporal_embed, spatial_embed, h_spatials):
    method forward (line 291) | def forward(self, h_temporal, h_spatials):
  class SRNN (line 326) | class SRNN(nn.Module):
    method __init__ (line 330) | def __init__(self, obs_space_dict, args, infer=False):
    method forward (line 389) | def forward(self, inputs, rnn_hxs, masks, infer=False):
  function reshapeT (line 471) | def reshapeT(T, seq_length, nenv):

FILE: rl/networks/storage.py
  function _flatten_helper (line 5) | def _flatten_helper(T, N, _tensor):
  class RolloutStorage (line 13) | class RolloutStorage(object):
    method __init__ (line 15) | def __init__(self, num_steps, num_processes, obs_shape, action_space,
    method to (line 56) | def to(self, device):
    method insert (line 70) | def insert(self, obs, recurrent_hidden_states, actions, action_log_probs,
    method after_update (line 88) | def after_update(self):
    method compute_returns (line 98) | def compute_returns(self,
    method feed_forward_generator (line 139) | def feed_forward_generator(self,
    method recurrent_generator (line 184) | def recurrent_generator(self, advantages, num_mini_batch):

FILE: rl/ppo/ppo.py
  class PPO (line 6) | class PPO():
    method __init__ (line 8) | def __init__(self,
    method update (line 36) | def update(self, rollouts):

FILE: rl/vec_env/dummy_vec_env.py
  class DummyVecEnv (line 6) | class DummyVecEnv(VecEnv):
    method __init__ (line 13) | def __init__(self, env_fns):
    method step_async (line 33) | def step_async(self, actions):
    method step_wait (line 47) | def step_wait(self):
    method reset (line 60) | def reset(self):
    method talk2Env_async (line 66) | def talk2Env_async(self, data):
    method talk2Env_wait (line 70) | def talk2Env_wait(self):
    method _save_obs (line 73) | def _save_obs(self, e, obs):
    method _obs_from_buf (line 83) | def _obs_from_buf(self):
    method get_images (line 86) | def get_images(self):
    method render (line 89) | def render(self, mode='human'):

FILE: rl/vec_env/envs.py
  function make_env (line 18) | def make_env(env_id, seed, rank):
  function make_vec_envs (line 33) | def make_vec_envs(env_name,
  class TimeLimitMask (line 62) | class TimeLimitMask(gym.Wrapper):
    method step (line 63) | def step(self, action):
    method reset (line 70) | def reset(self, **kwargs):
  class VecPyTorch (line 73) | class VecPyTorch(VecEnvWrapper):
    method __init__ (line 74) | def __init__(self, venv, device):
    method reset (line 80) | def reset(self):
    method step_async (line 89) | def step_async(self, actions):
    method step_wait (line 96) | def step_wait(self):

FILE: rl/vec_env/logger.py
  class KVWriter (line 19) | class KVWriter(object):
    method writekvs (line 20) | def writekvs(self, kvs):
  class SeqWriter (line 23) | class SeqWriter(object):
    method writeseq (line 24) | def writeseq(self, seq):
  class HumanOutputFormat (line 27) | class HumanOutputFormat(KVWriter, SeqWriter):
    method __init__ (line 28) | def __init__(self, filename_or_file):
    method writekvs (line 37) | def writekvs(self, kvs):
    method _truncate (line 71) | def _truncate(self, s):
    method writeseq (line 75) | def writeseq(self, seq):
    method close (line 84) | def close(self):
  class JSONOutputFormat (line 88) | class JSONOutputFormat(KVWriter):
    method __init__ (line 89) | def __init__(self, filename):
    method writekvs (line 92) | def writekvs(self, kvs):
    method close (line 99) | def close(self):
  class CSVOutputFormat (line 102) | class CSVOutputFormat(KVWriter):
    method __init__ (line 103) | def __init__(self, filename):
    method writekvs (line 108) | def writekvs(self, kvs):
    method close (line 135) | def close(self):
  class TensorBoardOutputFormat (line 139) | class TensorBoardOutputFormat(KVWriter):
    method __init__ (line 143) | def __init__(self, dir):
    method writekvs (line 158) | def writekvs(self, kvs):
    method close (line 169) | def close(self):
  function make_output_format (line 174) | def make_output_format(format, ev_dir, log_suffix=''):
  function logkv (line 193) | def logkv(key, val):
  function logkv_mean (line 201) | def logkv_mean(key, val):
  function logkvs (line 207) | def logkvs(d):
  function dumpkvs (line 214) | def dumpkvs():
  function getkvs (line 220) | def getkvs():
  function log (line 224) | def log(*args, level=INFO):
  function debug (line 230) | def debug(*args):
  function info (line 233) | def info(*args):
  function warn (line 236) | def warn(*args):
  function error (line 239) | def error(*args):
  function set_level (line 243) | def set_level(level):
  function set_comm (line 249) | def set_comm(comm):
  function get_dir (line 252) | def get_dir():
  function profile_kv (line 263) | def profile_kv(scopename):
  function profile (line 271) | def profile(n):
  function get_current (line 289) | def get_current():
  class Logger (line 296) | class Logger(object):
    method __init__ (line 301) | def __init__(self, dir, output_formats, comm=None):
    method logkv (line 311) | def logkv(self, key, val):
    method logkv_mean (line 314) | def logkv_mean(self, key, val):
    method dumpkvs (line 319) | def dumpkvs(self):
    method log (line 337) | def log(self, *args, level=INFO):
    method set_level (line 343) | def set_level(self, level):
    method set_comm (line 346) | def set_comm(self, comm):
    method get_dir (line 349) | def get_dir(self):
    method close (line 352) | def close(self):
    method _do_log (line 358) | def _do_log(self, args):
  function get_rank_without_mpi_import (line 363) | def get_rank_without_mpi_import():
  function configure (line 372) | def configure(dir=None, format_strs=None, comm=None, log_suffix=''):
  function _configure_default_logger (line 401) | def _configure_default_logger():
  function reset (line 405) | def reset():
  function scoped_configure (line 412) | def scoped_configure(dir=None, format_strs=None, comm=None):
  function _demo (line 423) | def _demo():
  function read_json (line 456) | def read_json(fname):
  function read_csv (line 464) | def read_csv(fname):
  function read_tb (line 468) | def read_tb(path):

FILE: rl/vec_env/running_mean_std.py
  class RunningMeanStd (line 6) | class RunningMeanStd(object):
    method __init__ (line 7) | def __init__(self, epsilon = 1e-4, shape = ()):
    method update (line 18) | def update(self, arr):
    method update_from_moments (line 24) | def update_from_moments(self, batch_mean, batch_var, batch_count):

FILE: rl/vec_env/shmem_vec_env.py
  class ShmemVecEnv (line 20) | class ShmemVecEnv(VecEnv):
    method __init__ (line 25) | def __init__(self, env_fns, spaces=None, context='spawn'):
    method reset (line 61) | def reset(self):
    method step_async (line 69) | def step_async(self, actions):
    method step_wait (line 75) | def step_wait(self):
    method talk2Env_async (line 81) | def talk2Env_async(self, data):
    method talk2Env_wait (line 87) | def talk2Env_wait(self):
    method close_extras (line 92) | def close_extras(self):
    method get_images (line 103) | def get_images(self, mode='human'):
    method _decode_obses (line 108) | def _decode_obses(self, obs):
  function _subproc_worker (line 124) | def _subproc_worker(pipe, parent_pipe, env_fn_wrapper, obs_bufs, obs_sha...

FILE: rl/vec_env/tile_images.py
  function tile_images (line 3) | def tile_images(img_nhwc):

FILE: rl/vec_env/util.py
  function copy_obs_dict (line 11) | def copy_obs_dict(obs):
  function dict_to_obs (line 18) | def dict_to_obs(obs_dict):
  function obs_space_info (line 28) | def obs_space_info(obs_space):
  function obs_to_dict (line 56) | def obs_to_dict(obs):

FILE: rl/vec_env/vec_env.py
  class AlreadySteppingError (line 10) | class AlreadySteppingError(Exception):
    method __init__ (line 16) | def __init__(self):
  class NotSteppingError (line 21) | class NotSteppingError(Exception):
    method __init__ (line 27) | def __init__(self):
  class VecEnv (line 32) | class VecEnv(ABC):
    method __init__ (line 46) | def __init__(self, num_envs, observation_space, action_space):
    method reset (line 52) | def reset(self):
    method step_async (line 64) | def step_async(self, actions):
    method step_wait (line 76) | def step_wait(self):
    method talk2Env_async (line 90) | def talk2Env_async(self, data):
    method talk2Env_wait (line 94) | def talk2Env_wait(self):
    method close_extras (line 97) | def close_extras(self):
    method close (line 104) | def close(self):
    method step (line 112) | def step(self, actions):
    method render (line 121) | def render(self, mode='human'):
    method talk2Env (line 132) | def talk2Env(self, data):
    method get_images (line 136) | def get_images(self):
    method unwrapped (line 143) | def unwrapped(self):
    method get_viewer (line 149) | def get_viewer(self):
  class VecEnvWrapper (line 155) | class VecEnvWrapper(VecEnv):
    method __init__ (line 161) | def __init__(self, venv, observation_space=None, action_space=None):
    method step_async (line 167) | def step_async(self, actions):
    method talk2Env_async (line 170) | def talk2Env_async(self, data):
    method talk2Env_wait (line 174) | def talk2Env_wait(self):
    method reset (line 178) | def reset(self):
    method step_wait (line 182) | def step_wait(self):
    method close (line 185) | def close(self):
    method render (line 188) | def render(self, mode='human'):
    method get_images (line 191) | def get_images(self):
    method __getattr__ (line 194) | def __getattr__(self, name):
  class VecEnvObservationWrapper (line 199) | class VecEnvObservationWrapper(VecEnvWrapper):
    method process (line 201) | def process(self, obs):
    method reset (line 204) | def reset(self):
    method step_wait (line 208) | def step_wait(self):
  class CloudpickleWrapper (line 212) | class CloudpickleWrapper(object):
    method __init__ (line 217) | def __init__(self, x):
    method __getstate__ (line 220) | def __getstate__(self):
    method __setstate__ (line 224) | def __setstate__(self, ob):
  function clear_mpi_env_vars (line 230) | def clear_mpi_env_vars():

FILE: rl/vec_env/vec_frame_stack.py
  class VecFrameStack (line 6) | class VecFrameStack(VecEnvWrapper):
    method __init__ (line 7) | def __init__(self, venv, nstack):
    method step_wait (line 17) | def step_wait(self):
    method reset (line 26) | def reset(self):

FILE: rl/vec_env/vec_normalize.py
  class VecNormalize (line 4) | class VecNormalize(VecEnvWrapper):
    method __init__ (line 10) | def __init__(self, venv, ob=True, ret=True, clipob=10., cliprew=10., g...
    method step_wait (line 26) | def step_wait(self):
    method _obfilt (line 36) | def _obfilt(self, obs):
    method reset (line 44) | def reset(self):

FILE: rl/vec_env/vec_pretext_normalize.py
  class VecPretextNormalize (line 14) | class VecPretextNormalize(VecEnvWrapper):
    method __init__ (line 22) | def __init__(self, venv, ob=False, ret=False, clipob=10., cliprew=10.,...
    method talk2Env_async (line 61) | def talk2Env_async(self, data):
    method talk2Env_wait (line 65) | def talk2Env_wait(self):
    method step_wait (line 69) | def step_wait(self):
    method _obfilt (line 77) | def _obfilt(self, obs):
    method reset (line 85) | def reset(self):
    method process_obs_rew (line 112) | def process_obs_rew(self, O, done, rews=0.):

FILE: rl/vec_env/vec_remove_dict_obs.py
  class VecExtractDictObs (line 3) | class VecExtractDictObs(VecEnvObservationWrapper):
    method __init__ (line 4) | def __init__(self, venv, key):
    method process (line 9) | def process(self, obs):

FILE: test.py
  function main (line 16) | def main():

FILE: train.py
  function main (line 23) | def main():