updated broken imports for new model modules

2026-08-15 12:05:18 +08:00 · 2018-04-10 10:51:11 +08:00
parent 5fbc7effa6
commit 96f46514d6
3 changed files with 7 additions and 8 deletions
@@ -10,7 +10,7 @@ import os
 import tensorboard_logger as tb_log

 from models import Pointnet2ClsMSG as Pointnet
-from models.Pointnet2Cls import model_fn_decorator
+from models.pointnet2_msg_cls import model_fn_decorator
 from data import ModelNet40Cls
 import utils.pytorch_utils as pt_utils
 import data.data_utils as d_utils