We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 229f369 commit 39ee8ebCopy full SHA for 39ee8eb
paddleslim/common/rl_controller/__init__.py
@@ -19,9 +19,7 @@
19
import parl
20
from .ddpg import *
21
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))
+ pass
25
26
from .lstm import *
27
from .utils import *
paddleslim/nas/rl_nas.py
@@ -76,6 +76,15 @@ def __init__(self,
76
self.save_controller = save_controller
77
self.load_controller = load_controller
78
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
+
88
cls = RLCONTROLLER.get(key.upper())
89
90
server_ip, server_port = server_addr
0 commit comments