SYMBOL INDEX (142 symbols across 11 files)

FILE: src/envs/__init__.py
  function build_env (line 21) | def build_env(params):

FILE: src/envs/ode.py
  class UnknownSymPyOperator (line 37) | class UnknownSymPyOperator(Exception):
  class InvalidPrefixExpression (line 41) | class InvalidPrefixExpression(Exception):
    method __init__ (line 42) | def __init__(self, data):
    method __str__ (line 45) | def __str__(self):
  function has_inf_nan (line 49) | def has_inf_nan(*args):
  function second_index (line 59) | def second_index(x, bal):
  function simplify (line 69) | def simplify(f, seconds):
  function expr_to_fun_real (line 93) | def expr_to_fun_real(x, fun, dimension):
  class Node (line 105) | class Node:
    method __init__ (line 106) | def __init__(self, value, children=None):
    method push_child (line 110) | def push_child(self, child):
    method prefix (line 113) | def prefix(self):
    method qtree_prefix (line 120) | def qtree_prefix(self):
    method infix (line 127) | def infix(self):
    method __len__ (line 139) | def __len__(self):
    method __str__ (line 145) | def __str__(self):
  class ODEEnvironment (line 150) | class ODEEnvironment(object):
    method __init__ (line 158) | def __init__(self, params):
    method get_integer (line 333) | def get_integer(self, cplex=False):
    method generate_leaf (line 356) | def generate_leaf(self, degree, index):
    method generate_ops (line 364) | def generate_ops(self, arity):
    method generate_dist (line 381) | def generate_dist(self, max_ops):
    method sample_next_pos (line 405) | def sample_next_pos(self, nb_empty, nb_ops):
    method generate_tree (line 425) | def generate_tree(self, nb_ops, degree, index=0):
    method generate_polynomial (line 447) | def generate_polynomial(
    method batch_sequences (line 489) | def batch_sequences(self, sequences):
    method write_int (line 508) | def write_int(self, val):
    method parse_int (line 524) | def parse_int(self, lst):
    method write_float (line 542) | def write_float(self, value, precision=None):
    method parse_float (line 554) | def parse_float(self, lst):
    method write_complex (line 592) | def write_complex(self, value, precision=None):
    method parse_complex (line 605) | def parse_complex(self, lst):
    method input_to_infix (line 620) | def input_to_infix(self, lst):
    method output_to_infix (line 637) | def output_to_infix(self, lst):
    method prefix_to_infix (line 641) | def prefix_to_infix(self, expr):
    method _sympy_to_prefix (line 672) | def _sympy_to_prefix(self, op, expr):
    method sympy_to_prefix (line 703) | def sympy_to_prefix(self, expr):
    method encode_expr (line 729) | def encode_expr(self, tree, cplx=False):
    method compute_gradient (line 746) | def compute_gradient(self, expr, point, degree):
    method gen_ode_system_convergence (line 758) | def gen_ode_system_convergence(self, return_system=False):
    method compute_gradient_control (line 878) | def compute_gradient_control(self, expr, point, degree, p):
    method gen_control (line 899) | def gen_control(self, return_system=False, skip_unstable=False):
    method compute_gradient_control_t (line 1072) | def compute_gradient_control_t(self, expr, point, degree, p):
    method compute_rank (line 1091) | def compute_rank(self, A, B, degree, p, val):
    method gen_control_t (line 1111) | def gen_control_t(self):
    method generate_cond_init (line 1218) | def generate_cond_init(self, max_delay, dimension, unariesexp, unaries...
    method gen_fourier_cond_init (line 1307) | def gen_fourier_cond_init(self):
    method create_train_iterator (line 1402) | def create_train_iterator(self, task, data_path, params):
    method create_test_iterator (line 1428) | def create_test_iterator(
    method register_args (line 1459) | def register_args(parser):
  class EnvDataset (line 1636) | class EnvDataset(Dataset):
    method __init__ (line 1637) | def __init__(self, env, task, train, params, path, size=None):
    method collate_fn (line 1709) | def collate_fn(self, elements):
    method init_rng (line 1721) | def init_rng(self):
    method get_worker_id (line 1741) | def get_worker_id(self):
    method __len__ (line 1751) | def __len__(self):
    method __getitem__ (line 1757) | def __getitem__(self, index):
    method read_sample (line 1768) | def read_sample(self, index):
    method generate_sample (line 1780) | def generate_sample(self):

FILE: src/evaluator.py
  function check_fourier_cond_init (line 28) | def check_fourier_cond_init(env, src, tgt, hyp):
  function idx_to_infix (line 143) | def idx_to_infix(env, idx, input=True):
  function compare_gramians (line 152) | def compare_gramians(env, tgt, hyp, tolerance, norm1=False):
  function check_gramian (line 204) | def check_gramian(env, src, tgt, hyp):
  function check_hypothesis (line 266) | def check_hypothesis(eq):
  class Evaluator (line 338) | class Evaluator(object):
    method __init__ (line 342) | def __init__(self, trainer):
    method run_all_evals (line 352) | def run_all_evals(self):
    method truncate_at (line 377) | def truncate_at(self, x, xlen):
    method enc_dec_step (line 402) | def enc_dec_step(self, data_type, task, scores):
    method enc_dec_step_beam_fast (line 536) | def enc_dec_step_beam_fast(self, data_type, task, scores, size=None):
  function convert_to_text (line 812) | def convert_to_text(batch, lengths, id2word, params):

FILE: src/logger.py
  class LogFormatter (line 13) | class LogFormatter:
    method __init__ (line 14) | def __init__(self):
    method format (line 17) | def format(self, record):
  function create_logger (line 30) | def create_logger(filepath, rank):

FILE: src/model/__init__.py
  function check_model_params (line 18) | def check_model_params(params):
  function build_modules (line 30) | def build_modules(env, params):

FILE: src/model/transformer.py
  function Embedding (line 23) | def Embedding(num_embeddings, embedding_dim, padding_idx=None):
  function create_sinusoidal_embeddings (line 31) | def create_sinusoidal_embeddings(n_pos, dim, out):
  function get_masks (line 42) | def get_masks(slen, lengths, causal):
  class MultiHeadAttention (line 64) | class MultiHeadAttention(nn.Module):
    method __init__ (line 68) | def __init__(self, n_heads, dim, dropout):
    method forward (line 81) | def forward(self, input, mask, kv=None, use_cache=False):
  class TransformerFFN (line 141) | class TransformerFFN(nn.Module):
    method __init__ (line 143) | def __init__(self, in_dim, dim_hidden, out_dim, dropout):
    method forward (line 149) | def forward(self, input):
  class TransformerModel (line 157) | class TransformerModel(nn.Module):
    method __init__ (line 161) | def __init__(self, params, id2word, is_encoder, with_output):
    method forward (line 222) | def forward(self, mode, **kwargs):
    method fwd (line 234) | def fwd(self, x, lengths, causal, src_enc=None, src_len=None, position...
    method predict (line 325) | def predict(self, tensor, pred_mask, y, get_scores):
    method generate (line 339) | def generate(self, src_enc, src_len, max_len=200, sample_temperature=N...
    method generate_beam (line 417) | def generate_beam(self, src_enc, src_len, beam_size, length_penalty, e...
  class BeamHypotheses (line 594) | class BeamHypotheses(object):
    method __init__ (line 596) | def __init__(self, n_hyp, max_len, length_penalty, early_stopping):
    method __len__ (line 607) | def __len__(self):
    method add (line 613) | def add(self, hyp, sum_logprobs):
    method is_done (line 627) | def is_done(self, best_sum_logprobs):

FILE: src/optim.py
  class Adam (line 16) | class Adam(optim.Optimizer):
    method __init__ (line 23) | def __init__(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-8, weig...
    method __setstate__ (line 42) | def __setstate__(self, state):
    method step (line 45) | def step(self, closure=None):
  class AdamInverseSqrtWithWarmup (line 92) | class AdamInverseSqrtWithWarmup(Adam):
    method __init__ (line 108) | def __init__(
    method get_lr_for_step (line 137) | def get_lr_for_step(self, num_updates):
    method step (line 143) | def step(self, closure=None):
  class AdamCosineWithWarmup (line 150) | class AdamCosineWithWarmup(Adam):
    method __init__ (line 167) | def __init__(
    method get_lr_for_step (line 202) | def get_lr_for_step(self, num_updates):
    method step (line 231) | def step(self, closure=None):
  function get_optimizer (line 238) | def get_optimizer(parameters, s):

FILE: src/slurm.py
  function sig_handler (line 20) | def sig_handler(signum, frame):
  function term_handler (line 32) | def term_handler(signum, frame):
  function init_signal_handler (line 37) | def init_signal_handler():
  function init_distributed_mode (line 46) | def init_distributed_mode(params):

FILE: src/trainer.py
  class Trainer (line 29) | class Trainer(object):
    method __init__ (line 33) | def __init__(self, modules, env, params):
    method set_parameters (line 164) | def set_parameters(self):
    method set_optimizers (line 179) | def set_optimizers(self):
    method init_amp (line 190) | def init_amp(self):
    method optimize (line 211) | def optimize(self, loss):
    method iter (line 257) | def iter(self):
    method print_stats (line 265) | def print_stats(self):
    method save_checkpoint (line 307) | def save_checkpoint(self, name, include_optimizers=True):
    method reload_checkpoint (line 336) | def reload_checkpoint(self):
    method save_periodic (line 390) | def save_periodic(self):
    method save_best_model (line 402) | def save_best_model(self, scores):
    method end_epoch (line 418) | def end_epoch(self, scores):
    method get_batch (line 454) | def get_batch(self, task):
    method export_data (line 478) | def export_data(self, task):
    method enc_dec_step (line 502) | def enc_dec_step(self, task):

FILE: src/utils.py
  class AttrDict (line 33) | class AttrDict(dict):
    method __init__ (line 34) | def __init__(self, *args, **kwargs):
  function bool_flag (line 39) | def bool_flag(s):
  function initialize_exp (line 51) | def initialize_exp(params):
  function get_dump_path (line 94) | def get_dump_path(params):
  function to_cuda (line 130) | def to_cuda(*args):
  class TimeoutError (line 139) | class TimeoutError(BaseException):
  function timeout (line 143) | def timeout(seconds=10, error_message=os.strerror(errno.ETIME)):

FILE: train.py
  function get_parser (line 26) | def get_parser():
  function main (line 270) | def main(params):