↓ 7 callersMethod__init__(self, vocab_size, block_size, n_layer=[12, 6], n_head=8, n_embd=1024,
embd_pdrop=0., resid_p
models/ar.py:223
↓ 3 callersFunctionposes_to_indices Tokenize pose (x, y) to indices. Args: poses: (x, y), [b, f, 2] Return: pose dicces: [b, f, 2]
modules/tokenizers/pose_tokenizer.py:3
↓ 2 callersMethodgenerate_gt_pose_gt_yaw(self, token, feature, pose, yaw, pose_gt, yaw_gt, vqvae_codebook, sampling_mtd = None, temperature_k = 1.0, t
models/model.py:179
↓ 1 callersFunctionloop_over_db_files(
rank=0,
workers=1,
split_db_list=None,
ego_save_dir='',
seq_save_dir='', nuplandb_wrappe
video_data_preprocess/create_nuplan_json.py:120