add _init_path.py for using multiple PCDet in the same environment

tools/_init_path.py

+import sys
+sys.path.insert(0, '../')
\ No newline at end of file

tools/test.py

+import _init_path
 import argparse
 import datetime
 import glob

tools/train.py

+import _init_path
 import argparse
 import datetime
 import glob
@@ -6,7 +7,6 @@ from pathlib import Path
 from test import repeat_eval_ckpt
 
 import torch
-import torch.distributed as dist
 import torch.nn as nn
 from tensorboardX import SummaryWriter
 
@@ -170,6 +170,11 @@ def main():
         merge_all_iters_to_one_epoch=args.merge_all_iters_to_one_epoch
     )
 
+    import pdb
+    pdb.set_trace()
+    if hasattr(train_set, 'use_shared_memory') and train_set.use_shared_memory:
+        train_set.clean_shared_memory()
+
     logger.info('**********************End training %s/%s(%s)**********************\n\n\n'
                 % (cfg.EXP_GROUP_PATH, cfg.TAG, args.extra_tag))
 
@@ -193,8 +198,6 @@ def main():
     logger.info('**********************End evaluation %s/%s(%s)**********************' %
                 (cfg.EXP_GROUP_PATH, cfg.TAG, args.extra_tag))
 
-    if hasattr(train_set, 'use_shared_memory') and train_set.use_shared_memory:
-        train_set.clean_shared_memory()
 
 if __name__ == '__main__':
     main()