23 lines
470 B
Python
23 lines
470 B
Python
import torch.nn as nn
|
|
|
|
|
|
class VFETemplate(nn.Module):
|
|
def __init__(self, model_cfg, **kwargs):
|
|
super().__init__()
|
|
self.model_cfg = model_cfg
|
|
|
|
def get_output_feature_dim(self):
|
|
raise NotImplementedError
|
|
|
|
def forward(self, **kwargs):
|
|
"""
|
|
Args:
|
|
**kwargs:
|
|
|
|
Returns:
|
|
batch_dict:
|
|
...
|
|
vfe_features: (num_voxels, C)
|
|
"""
|
|
raise NotImplementedError
|