Update.

Eaphan · Nov 21, 2023 · 7297759 · 7297759
1 parent 6ae2124
commit 7297759
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/playground/detection.3d/waymo/stemd/STEMD.waymo.resnet18_wide2x.cdn.epoch12/net.py b/playground/detection.3d/waymo/stemd/STEMD.waymo.resnet18_wide2x.cdn.epoch12/net.py
@@ -1,12 +1,12 @@
 from env import CustomTrainer, CustomWDDataset
 from modules.optimizer import AdamWMulti
-from voxel_detr import VoxelDETR
+from stemd import STEMD
 
-__all__ = ["build_model", "AdamWMulti", "CustomTrainer", "CustomWDDataset", "VoxelDETR"]
+__all__ = ["build_model", "AdamWMulti", "CustomTrainer", "CustomWDDataset", "STEMD"]
 
 
 def build_model(self, config):
 
-    model = VoxelDETR(config)
+    model = STEMD(config)
 
     return model
diff --git a/...resnet18_wide2x.cdn.epoch12/voxel_detr.py → ...aymo.resnet18_wide2x.cdn.epoch12/stemd.py b/...resnet18_wide2x.cdn.epoch12/voxel_detr.py → ...aymo.resnet18_wide2x.cdn.epoch12/stemd.py
@@ -41,7 +41,7 @@ def forward(self, x, edge_index):
         #h = h + x
         return h
 
-class VoxelDETR(nn.Module):
+class STEMD(nn.Module):
     def __init__(self, config):
         super().__init__()