Merge pull request #32 from Megvii-BaseDetection/update-version

[Enhance]: Update version.
Megvii-BaseDetection · Aug 17, 2022 · a030aae · a030aae
2 parents 72116c3 + 83d8281
commit a030aae
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -13,7 +13,7 @@ estimation. For more details, please refer to our [paper on Arxiv](https://arxiv
 ### Installation
 **Step 0.** Install [pytorch](https://pytorch.org/)(v1.9.0).
 
-**Step 1.** Install [MMDetection3D](https://github.com/open-mmlab/mmdetection3d)(v0.18.0).
+**Step 1.** Install [MMDetection3D](https://github.com/open-mmlab/mmdetection3d)(v1.0.0rc4).
 
 **Step 2.** Install requirements.
 ```shell

diff --git a/callbacks/ema.py b/callbacks/ema.py
@@ -74,12 +74,13 @@ def on_fit_start(self, trainer, pl_module):
 
         bn_model_list = list()
         bn_model_dist_group_list = list()
-        for model_ref in trainer.model.model.modules():
+        for model_ref in trainer.model.modules():
             if isinstance(model_ref, SyncBatchNorm):
                 bn_model_list.append(model_ref)
                 bn_model_dist_group_list.append(model_ref.process_group)
                 model_ref.process_group = None
-        trainer.ema_model = ModelEMA(trainer.model.model.cuda(), 0.9990)
+        trainer.ema_model = ModelEMA(trainer.model.module.module.model.cuda(),
+                                     0.9990)
 
         for bn_model, dist_group in zip(bn_model_list,
                                         bn_model_dist_group_list):

diff --git a/requirements.txt b/requirements.txt
@@ -3,7 +3,7 @@ numpy
 nuscenes-devkit
 opencv-python-headless
 pandas
-pytorch-lightning==1.5.10
+pytorch-lightning==1.6.0
 scikit-image
 scipy
 tensorboardX