From 031d89eeef5532e36687cb01de214fd198e7bfd8 Mon Sep 17 00:00:00 2001
From: LielinJiang <jianglielin@baidu.com>
Date: Tue, 11 Feb 2020 13:50:53 +0800
Subject: [PATCH] fix nas python3 bug

---
 slim/nas/README.md                   | 2 +-
 slim/nas/mobilenetv2_search_space.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/slim/nas/README.md b/slim/nas/README.md
index 2029261c..200ba80a 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 59fdcc45..2703e161 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)
 
-- 
GitLab