SYMBOL INDEX (491 symbols across 50 files) FILE: dataloader/eval_loader.py function collate_fn (line 13) | def collate_fn(batch): class Text2MotionDataset (line 19) | class Text2MotionDataset(data.Dataset): method __init__ (line 20) | def __init__(self, dataset_name, split, w_vectorizer, feat_bias = 5, m... method reset_max_len (line 127) | def reset_max_len(self, length): method inv_transform (line 133) | def inv_transform(self, data): method forward_transform (line 136) | def forward_transform(self, data): method __len__ (line 139) | def __len__(self): method __getitem__ (line 142) | def __getitem__(self, item): function DATALoader (line 195) | def DATALoader(dataset_name, split, function cycle (line 208) | def cycle(iterable): FILE: dataloader/tokenizer_loader.py class VQMotionDataset (line 10) | class VQMotionDataset(data.Dataset): method __init__ (line 11) | def __init__(self, dataset_name, feat_bias = 5, window_size = 64, unit... method inv_transform (line 77) | def inv_transform(self, data): method __len__ (line 80) | def __len__(self): method __getitem__ (line 83) | def __getitem__(self, item): function DATALoader (line 98) | def DATALoader(dataset_name, function cycle (line 110) | def cycle(iterable): FILE: dataloader/vqvae_loader.py class VQMotionDataset (line 10) | class VQMotionDataset(data.Dataset): method __init__ (line 11) | def __init__(self, dataset_name, window_size = 64, unit_length = 4): method inv_transform (line 62) | def inv_transform(self, data): method compute_sampling_prob (line 65) | def compute_sampling_prob(self) : method __len__ (line 71) | def __len__(self): method __getitem__ (line 74) | def __getitem__(self, item): function DATALoader (line 85) | def DATALoader(dataset_name, function cycle (line 98) | def cycle(iterable): FILE: eval.py function main (line 32) | def main( FILE: eval_vqvae.py function main (line 24) | def main(): FILE: finetune_motion.py function main (line 27) | def main(): function train (line 104) | def train( function validate (line 160) | def validate(fabric: L.Fabric, model: torch.nn.Module, val_data: np.ndar... function loss_fn (line 174) | def loss_fn(logits, targets): function get_batch (line 182) | def get_batch(fabric: L.Fabric, data: list): function load_datasets (line 206) | def load_datasets(): FILE: generate.py function generate (line 15) | def generate( function main (line 73) | def main( FILE: generate_batch.py function generate (line 15) | def generate( function main (line 73) | def main( FILE: generate_motion.py function main (line 27) | def main( FILE: lit_llama/adapter.py class LLaMAConfig (line 18) | class LLaMAConfig(llama.LLaMAConfig): class CausalSelfAttention (line 23) | class CausalSelfAttention(nn.Module): method __init__ (line 27) | def __init__(self, config: LLaMAConfig, block_idx: int) -> None: method forward (line 50) | def forward(self, x: torch.Tensor) -> torch.Tensor: class Block (line 102) | class Block(nn.Module): method __init__ (line 106) | def __init__(self, config: LLaMAConfig, block_idx: int) -> None: method forward (line 113) | def forward(self, x: torch.Tensor) -> torch.Tensor: class LLaMA (line 119) | class LLaMA(llama.LLaMA): method __init__ (line 123) | def __init__(self, config: LLaMAConfig) -> None: method from_name (line 139) | def from_name(cls, name: str): function mark_only_adapter_as_trainable (line 143) | def mark_only_adapter_as_trainable(model: LLaMA) -> None: function adapter_state_from_state_dict (line 149) | def adapter_state_from_state_dict(state_dict: dict) -> dict: FILE: lit_llama/indexed_dataset.py function __best_fitting_dtype (line 24) | def __best_fitting_dtype(vocab_size=None): function get_available_dataset_impl (line 31) | def get_available_dataset_impl(): function infer_dataset_impl (line 35) | def infer_dataset_impl(path): function make_builder (line 51) | def make_builder(out_file, impl, vocab_size=None): function make_dataset (line 58) | def make_dataset(path, impl, skip_warmup=False): function dataset_exists (line 75) | def dataset_exists(path, impl): function read_longs (line 82) | def read_longs(f, n): function write_longs (line 88) | def write_longs(f, a): function code (line 104) | def code(dtype): function index_file_path (line 111) | def index_file_path(prefix_path): function data_file_path (line 115) | def data_file_path(prefix_path): function create_doc_idx (line 119) | def create_doc_idx(sizes): class IndexedDataset (line 127) | class IndexedDataset(torch.utils.data.Dataset): method __init__ (line 131) | def __init__(self, path): method read_index (line 137) | def read_index(self, path): method read_data (line 155) | def read_data(self, path): method check_index (line 158) | def check_index(self, i): method __del__ (line 162) | def __del__(self): method __getitem__ (line 167) | def __getitem__(self, idx): method __len__ (line 191) | def __len__(self): method num_tokens (line 194) | def num_tokens(self, index): method size (line 197) | def size(self, index): method exists (line 201) | def exists(path): method supports_prefetch (line 207) | def supports_prefetch(self): class IndexedCachedDataset (line 211) | class IndexedCachedDataset(IndexedDataset): method __init__ (line 213) | def __init__(self, path): method supports_prefetch (line 219) | def supports_prefetch(self): method prefetch (line 222) | def prefetch(self, indices): method __getitem__ (line 247) | def __getitem__(self, idx): class IndexedDatasetBuilder (line 264) | class IndexedDatasetBuilder(object): method __init__ (line 275) | def __init__(self, out_file, dtype=np.int32): method add_item (line 284) | def add_item(self, tensor): method end_document (line 291) | def end_document(self): method merge_file_ (line 294) | def merge_file_(self, another_file): method finalize (line 319) | def finalize(self, index_file): function _warmup_mmap_file (line 334) | def _warmup_mmap_file(path): class MMapIndexedDataset (line 340) | class MMapIndexedDataset(torch.utils.data.Dataset): class Index (line 341) | class Index(object): method writer (line 345) | def writer(cls, path, dtype): method __init__ (line 390) | def __init__(self, path, skip_warmup=False): method __del__ (line 427) | def __del__(self): method dtype (line 432) | def dtype(self): method sizes (line 436) | def sizes(self): method doc_idx (line 440) | def doc_idx(self): method __getitem__ (line 444) | def __getitem__(self, i): method __len__ (line 447) | def __len__(self): method __init__ (line 450) | def __init__(self, path, skip_warmup=False): method __getstate__ (line 459) | def __getstate__(self): method __setstate__ (line 462) | def __setstate__(self, state): method _do_init (line 465) | def _do_init(self, path, skip_warmup): method __del__ (line 477) | def __del__(self): method __len__ (line 482) | def __len__(self): method __getitem__ (line 486) | def __getitem__(self, idx): method get (line 507) | def get(self, idx, offset=0, length=None): method sizes (line 522) | def sizes(self): method doc_idx (line 526) | def doc_idx(self): method get_doc_idx (line 529) | def get_doc_idx(self): method set_doc_idx (line 532) | def set_doc_idx(self, doc_idx_): method supports_prefetch (line 536) | def supports_prefetch(self): method exists (line 540) | def exists(path): class MMapIndexedDatasetBuilder (line 546) | class MMapIndexedDatasetBuilder(object): method __init__ (line 547) | def __init__(self, out_file, dtype=np.int64): method dtype (line 554) | def dtype(self): method add_item (line 557) | def add_item(self, np_array): method add_doc (line 562) | def add_doc(self, np_array, sizes): method end_document (line 568) | def end_document(self): method merge_file_ (line 571) | def merge_file_(self, another_file): method finalize (line 584) | def finalize(self, index_file): FILE: lit_llama/lora.py class LoRALayer (line 18) | class LoRALayer(): method __init__ (line 19) | def __init__( class MergedLinear (line 38) | class MergedLinear(nn.Linear, LoRALayer): method __init__ (line 40) | def __init__( method reset_parameters (line 79) | def reset_parameters(self): method zero_pad (line 86) | def zero_pad(self, x): method train (line 94) | def train(self, mode: bool = True): method eval (line 109) | def eval(self): method forward (line 124) | def forward(self, x: torch.Tensor): function mark_only_lora_as_trainable (line 142) | def mark_only_lora_as_trainable(model: nn.Module, bias: str = 'none') ->... function lora_state_dict (line 162) | def lora_state_dict(model: nn.Module, bias: str = 'none') -> Dict[str, t... class LoRAConfig (line 182) | class LoRAConfig: class CausalSelfAttention (line 188) | class CausalSelfAttention(llama.CausalSelfAttention): method __init__ (line 191) | def __init__(self, config: llama.LLaMAConfig) -> None: function lora (line 218) | def lora(r, alpha, dropout, enabled: bool = True): FILE: lit_llama/model.py class LLaMAConfig (line 16) | class LLaMAConfig: method from_name (line 24) | def from_name(cls, name: str) -> Self: class LLaMA (line 36) | class LLaMA(nn.Module): method __init__ (line 37) | def __init__(self, config: LLaMAConfig) -> None: method _init_weights (line 64) | def _init_weights(self, module: nn.Module) -> None: method forward (line 70) | def forward(self, idx: torch.Tensor) -> torch.Tensor: method from_name (line 88) | def from_name(cls, name: str) -> Self: class Block (line 92) | class Block(nn.Module): method __init__ (line 93) | def __init__(self, config: LLaMAConfig) -> None: method forward (line 100) | def forward(self, x: torch.Tensor) -> torch.Tensor: class CausalSelfAttention (line 106) | class CausalSelfAttention(nn.Module): method __init__ (line 107) | def __init__(self, config: LLaMAConfig) -> None: method forward (line 121) | def forward(self, x: torch.Tensor) -> torch.Tensor: class MLP (line 161) | class MLP(nn.Module): method __init__ (line 162) | def __init__(self, config: LLaMAConfig) -> None: method forward (line 174) | def forward(self, x: torch.Tensor) -> torch.Tensor: class RMSNorm (line 180) | class RMSNorm(nn.Module): method __init__ (line 187) | def __init__(self, size: int, dim: int = -1, eps: float = 1e-5) -> None: method forward (line 193) | def forward(self, x: torch.Tensor) -> torch.Tensor: function build_rope_cache (line 203) | def build_rope_cache(seq_len: int, n_elem: int, dtype: torch.dtype, devi... function apply_rope (line 234) | def apply_rope(x: torch.Tensor, rope_cache: torch.Tensor) -> torch.Tensor: FILE: lit_llama/quantization.py class Linear8bitLt (line 29) | class Linear8bitLt(bnb.nn.Linear8bitLt): method __init__ (line 36) | def __init__(self, *args, **kwargs): method _load_from_state_dict (line 42) | def _load_from_state_dict(self, local_state_dict, *args, **kwargs): method _quantize_weight (line 56) | def _quantize_weight(self, weight: torch.Tensor) -> None: class ColBlockQuantizedLinear (line 68) | class ColBlockQuantizedLinear(torch.nn.Module): method __init__ (line 69) | def __init__(self, in_features, out_features, bias: bool, *, bits, til... method pack_weight (line 87) | def pack_weight(self, weight): method get_weight (line 97) | def get_weight(self, dtype=torch.float): method forward (line 108) | def forward(self, inp): class GPTQQuantizer (line 115) | class GPTQQuantizer: method __init__ (line 121) | def __init__(self, linear_module, *, bits, perchannel=True, sym=False,... method quantize_weight (line 144) | def quantize_weight(x, scale, zero, maxq): method find_params_weight (line 149) | def find_params_weight(self, x): method collect_input_stats (line 187) | def collect_input_stats(self, _1, inp, _2): method quantize (line 203) | def quantize(self): FILE: lit_llama/tokenizer.py class Tokenizer (line 9) | class Tokenizer: method __init__ (line 12) | def __init__(self, model_path: Path) -> None: method vocab_size (line 19) | def vocab_size(self) -> int: method encode (line 22) | def encode( method decode (line 43) | def decode(self, tokens: torch.Tensor) -> str: method train (line 47) | def train(input: str, destination: str, vocab_size=32000) -> None: FILE: lit_llama/utils.py function save_model_checkpoint (line 17) | def save_model_checkpoint(fabric, model, file_path): class EmptyInitOnDevice (line 46) | class EmptyInitOnDevice(torch.overrides.TorchFunctionMode): method __init__ (line 47) | def __init__(self, device=None, dtype=None, quantization_mode=None): method __enter__ (line 81) | def __enter__(self): method __exit__ (line 87) | def __exit__(self, exc_type, exc_val, exc_tb): method __torch_function__ (line 92) | def __torch_function__(self, func, types, args=(), kwargs=None): class NotYetLoadedTensor (line 117) | class NotYetLoadedTensor: method __init__ (line 118) | def __init__(self, metatensor, archiveinfo, storageinfo, rebuild_args): method rebuild (line 125) | def rebuild( method _load_tensor (line 156) | def _load_tensor(self): method __torch_function__ (line 178) | def __torch_function__(cls, func, types, args=(), kwargs=None): method __getattr__ (line 188) | def __getattr__(self, name): method __repr__ (line 215) | def __repr__(self): class LazyLoadingUnpickler (line 219) | class LazyLoadingUnpickler(pickle.Unpickler): method __init__ (line 220) | def __init__(self, file, zipfile): method find_class (line 224) | def find_class(self, module, name): method persistent_load (line 230) | def persistent_load(self, pid): function lazy_load (line 239) | def lazy_load(fn): FILE: models/encdec.py class Encoder (line 4) | class Encoder(nn.Module): method __init__ (line 5) | def __init__(self, method forward (line 32) | def forward(self, x): class Decoder (line 35) | class Decoder(nn.Module): method __init__ (line 36) | def __init__(self, method forward (line 65) | def forward(self, x): FILE: models/evaluator_wrapper.py function build_models (line 8) | def build_models(opt): class EvaluatorModelWrapper (line 30) | class EvaluatorModelWrapper(object): method __init__ (line 32) | def __init__(self, opt): method get_co_embeddings (line 64) | def get_co_embeddings(self, word_embs, pos_ohot, cap_lens, motions, m_... method get_motion_embeddings (line 80) | def get_motion_embeddings(self, motions, m_lens): FILE: models/modules.py function init_weight (line 5) | def init_weight(m): class MovementConvEncoder (line 13) | class MovementConvEncoder(nn.Module): method __init__ (line 14) | def __init__(self, input_size, hidden_size, output_size): method forward (line 28) | def forward(self, inputs): class TextEncoderBiGRUCo (line 36) | class TextEncoderBiGRUCo(nn.Module): method __init__ (line 37) | def __init__(self, word_size, pos_size, hidden_size, output_size, devi... method forward (line 58) | def forward(self, word_embs, pos_onehot, cap_lens): class MotionEncoderBiGRUCo (line 76) | class MotionEncoderBiGRUCo(nn.Module): method __init__ (line 77) | def __init__(self, input_size, hidden_size, output_size, device): method forward (line 96) | def forward(self, inputs, m_lens): FILE: models/quantize_cnn.py class QuantizeEMAReset (line 6) | class QuantizeEMAReset(nn.Module): method __init__ (line 7) | def __init__(self, nb_code, code_dim, args): method reset_codebook (line 14) | def reset_codebook(self): method _tile (line 20) | def _tile(self, x): method init_codebook (line 31) | def init_codebook(self, x): method compute_perplexity (line 39) | def compute_perplexity(self, code_idx) : method update_codebook (line 50) | def update_codebook(self, x, code_idx): method preprocess (line 75) | def preprocess(self, x): method quantize (line 81) | def quantize(self, x): method dequantize (line 89) | def dequantize(self, code_idx): method forward (line 94) | def forward(self, x): class Quantizer (line 127) | class Quantizer(nn.Module): method __init__ (line 128) | def __init__(self, n_e, e_dim, beta): method forward (line 138) | def forward(self, z): method quantize (line 166) | def quantize(self, z): method dequantize (line 178) | def dequantize(self, indices): method preprocess (line 185) | def preprocess(self, x): class QuantizeReset (line 193) | class QuantizeReset(nn.Module): method __init__ (line 194) | def __init__(self, nb_code, code_dim, args): method reset_codebook (line 201) | def reset_codebook(self): method _tile (line 205) | def _tile(self, x): method init_codebook (line 216) | def init_codebook(self, x): method compute_perplexity (line 223) | def compute_perplexity(self, code_idx) : method update_codebook (line 233) | def update_codebook(self, x, code_idx): method preprocess (line 254) | def preprocess(self, x): method quantize (line 260) | def quantize(self, x): method dequantize (line 268) | def dequantize(self, code_idx): method forward (line 273) | def forward(self, x): class QuantizeEMA (line 301) | class QuantizeEMA(nn.Module): method __init__ (line 302) | def __init__(self, nb_code, code_dim, args): method reset_codebook (line 309) | def reset_codebook(self): method _tile (line 315) | def _tile(self, x): method init_codebook (line 326) | def init_codebook(self, x): method compute_perplexity (line 334) | def compute_perplexity(self, code_idx) : method update_codebook (line 345) | def update_codebook(self, x, code_idx): method preprocess (line 365) | def preprocess(self, x): method quantize (line 371) | def quantize(self, x): method dequantize (line 379) | def dequantize(self, code_idx): method forward (line 384) | def forward(self, x): FILE: models/resnet.py class nonlinearity (line 4) | class nonlinearity(nn.Module): method __init__ (line 5) | def __init__(self): method forward (line 8) | def forward(self, x): class ResConv1DBlock (line 12) | class ResConv1DBlock(nn.Module): method __init__ (line 13) | def __init__(self, n_in, n_state, dilation=1, activation='silu', norm=... method forward (line 49) | def forward(self, x): class Resnet1D (line 71) | class Resnet1D(nn.Module): method __init__ (line 72) | def __init__(self, n_in, n_depth, dilation_growth_rate=1, reverse_dila... method forward (line 81) | def forward(self, x): FILE: models/rotation2xyz.py class Rotation2xyz (line 11) | class Rotation2xyz: method __init__ (line 12) | def __init__(self, device, dataset='amass'): method __call__ (line 17) | def __call__(self, x, mask, pose_rep, translation, glob, FILE: models/smpl.py class SMPL (line 64) | class SMPL(_SMPLLayer): method __init__ (line 67) | def __init__(self, model_path=SMPL_MODEL_PATH, **kwargs): method forward (line 86) | def forward(self, *args, **kwargs): FILE: models/vqvae.py class VQVAE_251 (line 7) | class VQVAE_251(nn.Module): method __init__ (line 8) | def __init__(self, method preprocess (line 37) | def preprocess(self, x): method postprocess (line 43) | def postprocess(self, x): method encode (line 49) | def encode(self, x): method forward (line 60) | def forward(self, x): method forward_decoder (line 75) | def forward_decoder(self, x): class HumanVQVAE (line 86) | class HumanVQVAE(nn.Module): method __init__ (line 87) | def __init__(self, method encode (line 105) | def encode(self, x): method forward (line 110) | def forward(self, x): method forward_decoder (line 116) | def forward_decoder(self, x): FILE: options/get_eval_option.py function is_float (line 6) | def is_float(numStr): function is_number (line 19) | def is_number(numStr): function get_opt (line 27) | def get_opt(opt_path, device): FILE: options/option.py function get_args_parser (line 3) | def get_args_parser(): FILE: options/option_vqvae.py function get_args_parser (line 3) | def get_args_parser(): FILE: scripts/convert_checkpoint.py function convert_state_dict (line 20) | def convert_state_dict(state_dict: Dict[str, torch.Tensor], dtype: torch... function meta_weights_for_nano_model (line 66) | def meta_weights_for_nano_model( FILE: scripts/convert_hf_checkpoint.py function convert_hf_checkpoint (line 18) | def convert_hf_checkpoint( FILE: scripts/download.py function download_original (line 11) | def download_original(wd: str) -> None: function download_from_hub (line 22) | def download_from_hub(repo_id: Optional[str] = None, local_dir: str = "c... FILE: scripts/generate_dataset.py function prepare (line 15) | def prepare(split): function main (line 47) | def main(): FILE: scripts/prepare_motion.py function prepare (line 20) | def prepare( function prepare_sample (line 50) | def prepare_sample(example: dict, tokenizer: Tokenizer, max_length: int,... function tokenize (line 79) | def tokenize(tokenizer: Tokenizer, string: str, max_length: int, eos=Tru... function generate_prompt (line 83) | def generate_prompt(example): function main (line 100) | def main(): FILE: static/js/bulma-carousel.js function __webpack_require__ (line 16) | function __webpack_require__(moduleId) { function detectSupportsPassive (line 189) | function detectSupportsPassive() { function defineProperties (line 273) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _toConsumableArray (line 275) | function _toConsumableArray(arr) { if (Array.isArray(arr)) { for (var i ... function _classCallCheck (line 277) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function EventEmitter (line 280) | function EventEmitter() { function defineProperties (line 324) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 326) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Coordinate (line 329) | function Coordinate() { function defineProperties (line 473) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _defineProperty (line 475) | function _defineProperty(obj, key, value) { if (key in obj) { Object.def... function _classCallCheck (line 477) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function _possibleConstructorReturn (line 479) | function _possibleConstructorReturn(self, call) { if (!self) { throw new... function _inherits (line 481) | function _inherits(subClass, superClass) { if (typeof superClass !== "fu... function bulmaCarousel (line 504) | function bulmaCarousel(selector) { function _toConsumableArray (line 983) | function _toConsumableArray(arr) { if (Array.isArray(arr)) { for (var i ... function defineProperties (line 1019) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 1021) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function _possibleConstructorReturn (line 1023) | function _possibleConstructorReturn(self, call) { if (!self) { throw new... function _inherits (line 1025) | function _inherits(subClass, superClass) { if (typeof superClass !== "fu... function Autoplay (line 1042) | function Autoplay(slider) { function defineProperties (line 1209) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 1211) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Breakpoints (line 1216) | function Breakpoints(slider) { function defineProperties (line 1347) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _toConsumableArray (line 1349) | function _toConsumableArray(arr) { if (Array.isArray(arr)) { for (var i ... function _classCallCheck (line 1351) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Infinite (line 1354) | function Infinite(slider) { function defineProperties (line 1427) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 1429) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Loop (line 1434) | function Loop(slider) { function defineProperties (line 1494) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 1496) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Navigation (line 1502) | function Navigation(slider) { function defineProperties (line 1644) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 1646) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Pagination (line 1653) | function Pagination(slider) { function defineProperties (line 1804) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 1806) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Swipe (line 1812) | function Swipe(slider) { function defineProperties (line 1946) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 1948) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Transitioner (line 1954) | function Transitioner(slider) { function defineProperties (line 2047) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 2049) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Fade (line 2054) | function Fade(transitioner, slider) { function defineProperties (line 2182) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 2184) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Translate (line 2190) | function Translate(transitioner, slider) { FILE: static/js/bulma-slider.js function __webpack_require__ (line 16) | function __webpack_require__(moduleId) { function defineProperties (line 86) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 90) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function _possibleConstructorReturn (line 92) | function _possibleConstructorReturn(self, call) { if (!self) { throw new... function _inherits (line 94) | function _inherits(subClass, superClass) { if (typeof superClass !== "fu... function bulmaSlider (line 105) | function bulmaSlider(selector) { function defineProperties (line 281) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 283) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function EventEmitter (line 286) | function EventEmitter() { FILE: train_vqvae.py function update_lr_warm_up (line 21) | def update_lr_warm_up(optimizer, nb_iter, warmup_step, lr): function main (line 33) | def main(): FILE: utils/evaluate.py function truncate_output_to_eos (line 15) | def truncate_output_to_eos(output, eos_id): function pad_left (line 24) | def pad_left(x, max_len, pad_id): function plot (line 31) | def plot(tokens, net, dataname): function vqvae_evaluation (line 44) | def vqvae_evaluation(out_dir, val_loader, net, logger, writer, eval_wrap... function evaluation (line 159) | def evaluation(val_loader, net, model, logger, tokenizer, eval_wrapper, ... function euclidean_distance_matrix (line 261) | def euclidean_distance_matrix(matrix1, matrix2): function calculate_top_k (line 278) | def calculate_top_k(mat, top_k): function calculate_R_precision (line 293) | def calculate_R_precision(embedding1, embedding2, top_k, sum_all=False): function calculate_diversity (line 304) | def calculate_diversity(activation, diversity_times): function calculate_frechet_distance (line 315) | def calculate_frechet_distance(mu1, sigma1, mu2, sigma2, eps=1e-6): function calculate_activation_statistics (line 352) | def calculate_activation_statistics(activations): function calculate_frechet_feature_distance (line 358) | def calculate_frechet_feature_distance(feature_list1, feature_list2): FILE: utils/losses.py class ReConsLoss (line 4) | class ReConsLoss(nn.Module): method __init__ (line 5) | def __init__(self, recons_loss, nb_joints): method forward (line 22) | def forward(self, motion_pred, motion_gt) : method forward_vel (line 26) | def forward_vel(self, motion_pred, motion_gt) : FILE: utils/motion_process.py function recover_root_rot_pos (line 4) | def recover_root_rot_pos(data): function recover_from_rot (line 26) | def recover_from_rot(data, joints_num, skeleton): function recover_from_ric (line 43) | def recover_from_ric(data, joints_num): FILE: utils/quaternion.py function qinv (line 16) | def qinv(q): function qinv_np (line 23) | def qinv_np(q): function qnormalize (line 28) | def qnormalize(q): function qmul (line 33) | def qmul(q, r): function qrot (line 54) | def qrot(q, v): function qeuler (line 76) | def qeuler(q, order, epsilon=0, deg=True): function qmul_np (line 128) | def qmul_np(q, r): function qrot_np (line 134) | def qrot_np(q, v): function qeuler_np (line 140) | def qeuler_np(q, order, epsilon=0, use_gpu=False): function qfix (line 149) | def qfix(q): function euler2quat (line 169) | def euler2quat(e, order, deg=True): function expmap_to_quaternion (line 214) | def expmap_to_quaternion(e): function euler_to_quaternion (line 233) | def euler_to_quaternion(e, order): function quaternion_to_matrix (line 274) | def quaternion_to_matrix(quaternions): function quaternion_to_matrix_np (line 303) | def quaternion_to_matrix_np(quaternions): function quaternion_to_cont6d_np (line 308) | def quaternion_to_cont6d_np(quaternions): function quaternion_to_cont6d (line 314) | def quaternion_to_cont6d(quaternions): function cont6d_to_matrix (line 320) | def cont6d_to_matrix(cont6d): function cont6d_to_matrix_np (line 339) | def cont6d_to_matrix_np(cont6d): function qpow (line 344) | def qpow(q0, t, dtype=torch.float): function qslerp (line 369) | def qslerp(q0, q1, t): function qbetween (line 387) | def qbetween(v0, v1): function qbetween_np (line 400) | def qbetween_np(v0, v1): function lerp (line 412) | def lerp(p0, p1, t): FILE: utils/rotation_conversions.py function quaternion_to_matrix (line 32) | def quaternion_to_matrix(quaternions): function _copysign (line 61) | def _copysign(a, b): function _sqrt_positive_part (line 77) | def _sqrt_positive_part(x): function matrix_to_quaternion (line 88) | def matrix_to_quaternion(matrix): function _axis_angle_rotation (line 111) | def _axis_angle_rotation(axis: str, angle): function euler_angles_to_matrix (line 137) | def euler_angles_to_matrix(euler_angles, convention: str): function _angle_from_tan (line 160) | def _angle_from_tan( function _index_from_letter (line 191) | def _index_from_letter(letter: str): function matrix_to_euler_angles (line 200) | def matrix_to_euler_angles(matrix, convention: str): function random_quaternions (line 240) | def random_quaternions( function random_rotations (line 262) | def random_rotations( function random_rotation (line 283) | def random_rotation( function standardize_quaternion (line 300) | def standardize_quaternion(quaternions): function quaternion_raw_multiply (line 313) | def quaternion_raw_multiply(a, b): function quaternion_multiply (line 332) | def quaternion_multiply(a, b): function quaternion_invert (line 347) | def quaternion_invert(quaternion): function quaternion_apply (line 361) | def quaternion_apply(quaternion, point): function axis_angle_to_matrix (line 382) | def axis_angle_to_matrix(axis_angle): function matrix_to_axis_angle (line 396) | def matrix_to_axis_angle(matrix): function axis_angle_to_quaternion (line 410) | def axis_angle_to_quaternion(axis_angle): function quaternion_to_axis_angle (line 440) | def quaternion_to_axis_angle(quaternions): function rotation_6d_to_matrix (line 469) | def rotation_6d_to_matrix(d6: torch.Tensor) -> torch.Tensor: function matrix_to_rotation_6d (line 491) | def matrix_to_rotation_6d(matrix: torch.Tensor) -> torch.Tensor: function canonicalize_smplh (line 506) | def canonicalize_smplh(poses, trans = None): FILE: utils/skeleton.py class Skeleton (line 4) | class Skeleton(object): method __init__ (line 5) | def __init__(self, offset, kinematic_tree, device): method njoints (line 17) | def njoints(self): method offset (line 20) | def offset(self): method set_offset (line 23) | def set_offset(self, offsets): method kinematic_tree (line 26) | def kinematic_tree(self): method parents (line 29) | def parents(self): method get_offsets_joints_batch (line 33) | def get_offsets_joints_batch(self, joints): method get_offsets_joints (line 43) | def get_offsets_joints(self, joints): method inverse_kinematics_np (line 55) | def inverse_kinematics_np(self, joints, face_joint_idx, smooth_forward... method forward_kinematics (line 104) | def forward_kinematics(self, quat_params, root_pos, skel_joints=None, ... method forward_kinematics_np (line 126) | def forward_kinematics_np(self, quat_params, root_pos, skel_joints=Non... method forward_kinematics_cont6d_np (line 149) | def forward_kinematics_cont6d_np(self, cont6d_params, root_pos, skel_j... method forward_kinematics_cont6d (line 173) | def forward_kinematics_cont6d(self, cont6d_params, root_pos, skel_join... FILE: utils/utils_model.py function getCi (line 8) | def getCi(accLog): function get_logger (line 16) | def get_logger(out_dir): function initial_optim (line 33) | def initial_optim(decay_option, lr, weight_decay, net, optimizer) : function get_motion_with_trans (line 55) | def get_motion_with_trans(motion, velocity) : FILE: utils/word_vectorizer.py class WordVectorizer (line 46) | class WordVectorizer(object): method __init__ (line 47) | def __init__(self, meta_root, prefix): method _get_pos_ohot (line 53) | def _get_pos_ohot(self, pos): method __len__ (line 61) | def __len__(self): method __getitem__ (line 64) | def __getitem__(self, item): class WordVectorizerV2 (line 83) | class WordVectorizerV2(WordVectorizer): method __init__ (line 84) | def __init__(self, meta_root, prefix): method __getitem__ (line 88) | def __getitem__(self, item): method itos (line 96) | def itos(self, idx): FILE: visualization/plot_3d_global.py function plot_3d_motion (line 12) | def plot_3d_motion(args, figsize=(10, 10), fps=120, radius=4): function draw_to_batch (line 114) | def draw_to_batch(smpl_joints_batch, title_batch=None, outname=None) : FILE: visualization/render.py class WeakPerspectiveCamera (line 20) | class WeakPerspectiveCamera(pyrender.Camera): method __init__ (line 21) | def __init__(self, method get_projection_matrix (line 35) | def get_projection_matrix(self, width=None, height=None): function render (line 44) | def render(motions, name, outdir='render_vis', device_id=0): FILE: visualize/joints2smpl/src/customloss.py function gmof (line 6) | def gmof(x, sigma): function angle_prior (line 15) | def angle_prior(pose): function perspective_projection (line 24) | def perspective_projection(points, rotation, translation, function body_fitting_loss (line 55) | def body_fitting_loss(body_pose, betas, model_joints, camera_t, camera_c... function camera_fitting_loss (line 91) | def camera_fitting_loss(model_joints, camera_t, camera_t_est, camera_cen... function body_fitting_loss_3d (line 128) | def body_fitting_loss_3d(body_pose, preserve_pose, function camera_fitting_loss_3d (line 192) | def camera_fitting_loss_3d(model_joints, camera_t, camera_t_est, FILE: visualize/joints2smpl/src/prior.py function create_prior (line 35) | def create_prior(prior_type, **kwargs): class SMPLifyAnglePrior (line 52) | class SMPLifyAnglePrior(nn.Module): method __init__ (line 53) | def __init__(self, dtype=torch.float32, **kwargs): method forward (line 72) | def forward(self, pose, with_global_pose=False): class L2Prior (line 91) | class L2Prior(nn.Module): method __init__ (line 92) | def __init__(self, dtype=DEFAULT_DTYPE, reduction='sum', **kwargs): method forward (line 95) | def forward(self, module_input, *args): class MaxMixturePrior (line 99) | class MaxMixturePrior(nn.Module): method __init__ (line 101) | def __init__(self, prior_folder='prior', method get_mean (line 175) | def get_mean(self): method merged_log_likelihood (line 180) | def merged_log_likelihood(self, pose, betas): method log_likelihood (line 197) | def log_likelihood(self, pose, betas, *args, **kwargs): method forward (line 226) | def forward(self, pose, betas): FILE: visualize/joints2smpl/src/smplify.py function guess_init_3d (line 19) | def guess_init_3d(model_joints, class SMPLify3D (line 44) | class SMPLify3D(): method __init__ (line 47) | def __init__(self, method __call__ (line 95) | def __call__(self, init_pose, init_betas, init_cam_t, j3d, conf_3d=1.0... FILE: visualize/simplify_loc2rot.py class joints2smpl (line 13) | class joints2smpl: method __init__ (line 15) | def __init__(self, num_frames, device_id, cuda=True): method npy2smpl (line 45) | def npy2smpl(self, npy_path): method joint2smpl (line 63) | def joint2smpl(self, input_joints, init_params=None): FILE: visualize/vis_utils.py class npy2obj (line 8) | class npy2obj: method __init__ (line 9) | def __init__(self, npy_path, sample_idx, rep_idx, device=0, cuda=True): method get_vertices (line 43) | def get_vertices(self, sample_i, frame_i): method get_trimesh (line 46) | def get_trimesh(self, sample_i, frame_i): method save_obj (line 50) | def save_obj(self, save_path, frame_i): method save_npy (line 56) | def save_npy(self, save_path):