diff --git a/slim/nas/README.md b/slim/nas/README.md
index 2029261cb70e55af919a815ece0229bc1ffac530..200ba80ac506d82570f5881bfd4fced2c95a86c9 100644
--- a/slim/nas/README.md
+++ b/slim/nas/README.md
@@ -46,7 +46,7 @@ SLIM:
 ## 训练与评估
 执行以下命令，边训练边评估
 ```shell
-python -u ./slim/nas/train_nas.py --log_steps 10 --cfg configs/deeplabv3p_mobilenetv2_cityscapes.yaml --use_gpu --use_mpio \
+CUDA_VISIBLE_DEVICES=0 python -u ./slim/nas/train_nas.py --log_steps 10 --cfg configs/deeplabv3p_mobilenetv2_cityscapes.yaml --use_gpu --use_mpio \
 SLIM.NAS_PORT 23333 \
 SLIM.NAS_ADDRESS "" \
 SLIM.NAS_SEARCH_STEPS 2 \
diff --git a/slim/nas/mobilenetv2_search_space.py b/slim/nas/mobilenetv2_search_space.py
index 59fdcc45374393c550b6f8dff5f21ca98cd70302..2703e161f02e9659040b827fff8d345db5bf5946 100644
--- a/slim/nas/mobilenetv2_search_space.py
+++ b/slim/nas/mobilenetv2_search_space.py
@@ -180,7 +180,7 @@ class MobileNetV2SpaceSeg(SearchSpaceBase):
                     c=int(c * self.scale),
                     n=n,
                     s=s,
-                    k=k,
+                    k=int(k),
                     name='mobilenetv2_conv' + str(i))
                 in_c = int(c * self.scale)