SYMBOL INDEX (111 symbols across 11 files)

FILE: data.py
  class PoseDataset (line 16) | class PoseDataset(torch.utils.data.Dataset):
    method __init__ (line 17) | def __init__(self, data_dir, tolerance=False):
    method __getitem__ (line 71) | def __getitem__(self, index):
    method __len__ (line 106) | def __len__(self):
  class MovementAudDataset (line 110) | class MovementAudDataset(torch.utils.data.Dataset):
    method __init__ (line 111) | def __init__(self, data_dir):
    method __getitem__ (line 155) | def __getitem__(self, index):
    method __len__ (line 185) | def __len__(self):
  function get_loader (line 188) | def get_loader(batch_size, shuffle, num_workers, dataset, data_dir, tole...

FILE: demo.py
  function loadDecompModel (line 28) | def loadDecompModel(args):
  function loadCompModel (line 40) | def loadCompModel(args):

FILE: model_comp.py
  class Trainer_Comp (line 26) | class Trainer_Comp(object):
    method __init__ (line 27) | def __init__(self, data_loader, dance_enc, dance_dec, danceAud_dis, mo...
    method init_logs (line 65) | def init_logs(self):
    method get_z_random (line 73) | def get_z_random(self, batchSize, nz, random_type='gauss'):
    method ones_like (line 78) | def ones_like(tensor, val=1.):
    method zeros_like (line 82) | def zeros_like(tensor, val=0.):
    method kld_coef (line 84) | def kld_coef(self, i):
    method forward (line 88) | def forward(self, stdpSeq, batchsize, aud_style, aud):
    method backward_D (line 138) | def backward_D(self):
    method backward_danceED (line 167) | def backward_danceED(self):
    method backward_info_ondance (line 199) | def backward_info_ondance(self):
    method zero_grad (line 206) | def zero_grad(self, opt_list):
    method clip_norm (line 210) | def clip_norm(self, network_list):
    method step (line 214) | def step(self, opt_list):
    method update (line 218) | def update(self):
    method test_final (line 237) | def test_final(self, initpose, aud, n, thr=0):
    method resume (line 282) | def resume(self, model_dir, train=True):
    method save (line 300) | def save(self, filename, ep, total_it):
    method cuda (line 324) | def cuda(self):
    method train (line 338) | def train(self, ep=0, it=0):

FILE: model_decomp.py
  class Trainer_Decomp (line 26) | class Trainer_Decomp(object):
    method __init__ (line 27) | def __init__(self, data_loader, initp_enc, initp_dec, movement_enc, st...
    method init_logs (line 53) | def init_logs(self):
    method get_z_random (line 59) | def get_z_random(self, batchSize, nz, random_type='gauss'):
    method ones_like (line 64) | def ones_like(tensor, val=1.):
    method zeros_like (line 68) | def zeros_like(tensor, val=0.):
    method random_generate_stdp (line 72) | def random_generate_stdp(self, init_p):
    method forward (line 82) | def forward(self, stdpose1, stdpose2):
    method backward_initp_ED (line 128) | def backward_initp_ED(self):
    method backward_movement_ED (line 138) | def backward_movement_ED(self):
    method update (line 160) | def update(self):
    method save (line 177) | def save(self, filename, ep, total_it):
    method resume (line 193) | def resume(self, model_dir, train=True):
    method kld_coef (line 208) | def kld_coef(self, i):
    method generate_stdp_sequence (line 212) | def generate_stdp_sequence(self, initpose, aud, num_stdp):
    method cuda (line 258) | def cuda(self):
    method train (line 265) | def train(self, ep=0, it=0):

FILE: modulate.py
  function modulate (line 14) | def modulate(dance, beats, length):
  function get_pose (line 56) | def get_pose(pose, n):

FILE: networks.py
  class InitPose_Enc (line 26) | class InitPose_Enc(nn.Module):
    method __init__ (line 27) | def __init__(self, pose_size, dim_z_init):
    method forward (line 45) | def forward(self, pose):
  class InitPose_Dec (line 49) | class InitPose_Dec(nn.Module):
    method __init__ (line 50) | def __init__(self, pose_size, dim_z_init):
    method forward (line 63) | def forward(self, z_init):
  class Movement_Enc (line 66) | class Movement_Enc(nn.Module):
    method __init__ (line 67) | def __init__(self, pose_size, dim_z_movement, length, hidden_size, num...
    method forward (line 98) | def forward(self, poses):
    method getFeature (line 110) | def getFeature(self, poses):
  class StandardPose_Dec (line 120) | class StandardPose_Dec(nn.Module):
    method __init__ (line 121) | def __init__(self, pose_size, dim_z_init, dim_z_movement, length, hidd...
    method forward (line 147) | def forward(self, z_init, z_movement):
  class StandardPose_Dis (line 158) | class StandardPose_Dis(nn.Module):
    method __init__ (line 159) | def __init__(self, pose_size, length):
    method forward (line 176) | def forward(self, pose_seq):
  class Dance_Enc (line 185) | class Dance_Enc(nn.Module):
    method __init__ (line 186) | def __init__(self, dim_z_movement, dim_z_dance, hidden_size, num_layer...
    method forward (line 210) | def forward(self, movements_mean, movements_std):
  class Dance_Dec (line 221) | class Dance_Dec(nn.Module):
    method __init__ (line 222) | def __init__(self, dim_z_dance, dim_z_movement, hidden_size, num_layers):
    method forward (line 254) | def forward(self, z_dance, length=3):
  class DanceAud_Dis2 (line 266) | class DanceAud_Dis2(nn.Module):
    method __init__ (line 267) | def __init__(self, aud_size, dim_z_movement, length=3):
    method forward (line 299) | def forward(self, movements, aud):
  class DanceAud_Dis (line 308) | class DanceAud_Dis(nn.Module):
    method __init__ (line 309) | def __init__(self, aud_size, dim_z_movement, length=3):
    method forward (line 331) | def forward(self, movements, aud):
  class DanceAud_InfoDis (line 340) | class DanceAud_InfoDis(nn.Module):
    method __init__ (line 341) | def __init__(self, aud_size, dim_z_movement, length):
    method forward (line 370) | def forward(self, movements, aud):
  class Dance2Style (line 375) | class Dance2Style(nn.Module):
    method __init__ (line 376) | def __init__(self, dim_z_dance, aud_size):
    method forward (line 393) | def forward(self, zdance):
  class AudioClassifier_rnn (line 401) | class AudioClassifier_rnn(nn.Module):
    method __init__ (line 402) | def __init__(self, dim_z_motion, hidden_size, pose_size, cls, num_laye...
    method forward (line 419) | def forward(self, poses):
    method get_style (line 424) | def get_style(self, auds):
  class Audstyle_Enc (line 430) | class Audstyle_Enc(nn.Module):
    method __init__ (line 431) | def __init__(self, aud_size, dim_z, dim_noise=30):
    method forward (line 450) | def forward(self, aud):

FILE: options.py
  class DecompOptions (line 11) | class DecompOptions():
    method __init__ (line 12) | def __init__(self):
    method parse (line 51) | def parse(self):
  class CompOptions (line 56) | class CompOptions():
    method __init__ (line 57) | def __init__(self):
    method parse (line 103) | def parse(self):
  class TestOptions (line 108) | class TestOptions():
    method __init__ (line 109) | def __init__(self):
    method parse (line 160) | def parse(self):

FILE: test.py
  function loadDecompModel (line 37) | def loadDecompModel(args):
  function loadCompModel (line 49) | def loadCompModel(args):

FILE: train_comp.py
  function loadDecompModel (line 21) | def loadDecompModel(args):
  function getCompNetworks (line 33) | def getCompNetworks(args):

FILE: train_decomp.py
  function getDecompNetworks (line 21) | def getDecompNetworks(args):

FILE: utils.py
  class Logger (line 16) | class Logger(object):
    method __init__ (line 17) | def __init__(self, log_dir):
    method scalar_summary (line 20) | def scalar_summary(self, tag, value, step):
  function vis (line 24) | def vis(poses, outdir, aud=None):
  function vis2 (line 73) | def vis2(poses, outdir, fibeat):
  function vis_single (line 126) | def vis_single(pose, outfile):