↓ 1 callersFunctionpk_sampling(logits, top_k=30, temperature_k=1.0, top_p=0.9, temperature_p=1.0, filter_value=-float('Inf'))
utils/generat_utils.py:47
↓ 1 callersMethodstep_eval(self, token, feature, pose, yaw, token_gt, feature_gt, pose_gt, yaw_gt, vqvae_codebook)
models/model.py:122
↓ 1 callersMethodstep_train(self, token, feature, pose, yaw, token_gt, feature_gt, pose_gt, yaw_gt, vqvae_codebook=None)
models/model.py:84
Method__init__(self, in_channels=3, ch=128, ch_mult=(1,1,2,2,4), num_res_blocks=2,
norm_type='group', drop
modules/tokenizers/vq_model.py:160
Method__init__(self, z_channels=256, ch=128, ch_mult=(1,1,2,2,4), num_res_blocks=2, norm_type="group",
drop
modules/tokenizers/vq_model.py:208
Method__init__(self, n_e, e_dim, beta, entropy_loss_ratio, l2_norm, show_usage)
modules/tokenizers/vq_model.py:260
Method__init__(self, in_channels, out_channels=None, conv_shortcut=False, dropout=0.0, norm_type='group')
modules/tokenizers/vq_model.py:358
Method__init__(self, n_heads, dim, multiple_of=256, ffn_dim_multiplier=None, norm_eps=1e-5)
modules/tokenizers/vq_model.py:433
Method__init__(self, data_root, json_root, condition_frames=3, downsample_fps=5, downsample_size=16, h=256, w=512)
datasets/dataset_nuplan.py:12
Method__setstate__(self, state: Tuple[dict, Optional[str], Optional[str],
dict, bool, se
utils/config_utils.py:2044