Method__init__(self, dim, num_heads=8, qkv_bias=False, attn_drop=0.2, proj_drop=0.2)
model/transformer_branch.py:75
Method__init__(self, num_classes, embed_dim, use_att, dropout=0.1,
extra_feature_channels=3, width_multipli
model/pvcnn_generation.py:174
Method__init__(self, root_pc:str, root_views: str, cache: str, categories: list = ['chair'], split: str= 'val',
datasets/shapenet_data_sv.py:52
Functionavg_voxelize_forwardFunction: average pool voxelization (forward)
Args:
features: features, FloatTensor[b, c, n]
coords : coords of each point, IntTensor[b, 3,
modules/functional/src/voxelization/vox.cpp:17
Methodbackward :param ctx: :param grad_output: gradient of output, FloatTensor[B, C, R, R, R] :return: gradient of inputs, Float
modules/functional/voxelization.py:27
Functioncreate_pointnet_components(blocks, in_channels, embed_dim, with_se=False, normalize=True, eps=0,
width_mu
model/pvcnn_generation.py:46