Commit 007f2e68 authored by 雍大凯's avatar 雍大凯
Browse files

将子模块转换为普通目录

parent 19472568
from .bevformer import BEVFormer
from .bevformer_fp16 import BEVFormer_fp16
from .bevformerV2 import BEVFormerV2
\ No newline at end of file
from .custom_hooks import TransferWeight
\ No newline at end of file
from mmcv.runner.hooks.hook import HOOKS, Hook
from projects.mmdet3d_plugin.models.utils import run_time
@HOOKS.register_module()
class TransferWeight(Hook):
def __init__(self, every_n_inters=1):
self.every_n_inters=every_n_inters
def after_train_iter(self, runner):
if self.every_n_inner_iters(runner, self.every_n_inters):
runner.eval_model.load_state_dict(runner.model.state_dict())
from .transformer import PerceptionTransformer
from .transformerV2 import PerceptionTransformerV2, PerceptionTransformerBEVEncoder
from .spatial_cross_attention import SpatialCrossAttention, MSDeformableAttention3D
from .temporal_self_attention import TemporalSelfAttention
from .encoder import BEVFormerEncoder, BEVFormerLayer
from .decoder import DetectionTransformerDecoder
from .group_attention import GroupMultiheadAttention
from .epoch_based_runner import EpochBasedRunner_video
\ No newline at end of file
from .hungarian_assigner_3d import HungarianAssigner3D
__all__ = ['HungarianAssigner3D']
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment