diff --git a/configs/softgroup_scannet.yaml b/configs/softgroup_scannet.yaml
index b1e9dda..51c129e 100644
--- a/configs/softgroup_scannet.yaml
+++ b/configs/softgroup_scannet.yaml
@@ -66,8 +66,9 @@ optimizer:
   type: 'Adam'
   lr: 0.004
 
+fp16: False
 epochs: 128
 step_epoch: 50
 save_freq: 4
-pretrain: 'work_dirs/softgroup_scannet_backbone_spconv2_dist/epoch_116.pth'
-work_dir: 'work_dirs/softgroup_scannet_spconv2_dist'
+pretrain: 'work_dirs/softgroup_scannet_backbone/epoch_120.pth'
+work_dir: ''
diff --git a/configs/softgroup_scannet_backbone.yaml b/configs/softgroup_scannet_backbone.yaml
index fb354a6..2e49c89 100644
--- a/configs/softgroup_scannet_backbone.yaml
+++ b/configs/softgroup_scannet_backbone.yaml
@@ -71,4 +71,4 @@ epochs: 128
 step_epoch: 50
 save_freq: 4
 pretrain: ''
-work_dir: 'work_dirs/softgroup_scannet_backbone'
+work_dir: ''
diff --git a/requirements.txt b/requirements.txt
index 466b18e..c184ea4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,9 +1,10 @@
-cmake>=3.13.2
+munch
 pandas
 plyfile
 pyyaml==5.4.1
 scikit-learn
 scipy
 six
+tensorboard
 tensorboardX
-torch==1.1
+tqdm
diff --git a/test.py b/test.py
index 50ae660..c492bd4 100644
--- a/test.py
+++ b/test.py
@@ -33,10 +33,9 @@ if __name__ == '__main__':
     cfg = Munch.fromDict(yaml.safe_load(cfg_txt))
     logger = get_root_logger()
 
-    model = SoftGroup(**cfg.model)
+    model = SoftGroup(**cfg.model).cuda()
     logger.info(f'Load state dict from {args.checkpoint}')
     load_checkpoint(args.checkpoint, logger, model)
-    model.cuda()
 
     dataset = build_dataset(cfg.data.test, logger)
     dataloader = build_dataloader(dataset, training=False, **cfg.dataloader.test)