Skip to content

Commit 39ee8eb

Browse files
authored
Fix warning (#345)
* fix * refine warning
1 parent 229f369 commit 39ee8eb

File tree

2 files changed

+10
-3
lines changed

2 files changed

+10
-3
lines changed

paddleslim/common/rl_controller/__init__.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,7 @@
1919
import parl
2020
from .ddpg import *
2121
except ImportError as e:
22-
_logger.warn(
23-
"If you want to use DDPG in RLNAS, please pip install parl first. Now states: {}".
24-
format(e))
22+
pass
2523

2624
from .lstm import *
2725
from .utils import *

paddleslim/nas/rl_nas.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,15 @@ def __init__(self,
7676
self.save_controller = save_controller
7777
self.load_controller = load_controller
7878

79+
if key.upper() in ['DDPG']:
80+
try:
81+
import parl
82+
except ImportError as e:
83+
_logger.error(
84+
"If you want to use DDPG in RLNAS, please pip install parl first. Now states: {}".
85+
format(e))
86+
os._exit(1)
87+
7988
cls = RLCONTROLLER.get(key.upper())
8089

8190
server_ip, server_port = server_addr

0 commit comments

Comments
 (0)