未验证 提交 6457bba2 编写于 作者: C ceci3 提交者: GitHub

fix nas (#553)

上级 18f80c7e
...@@ -75,6 +75,7 @@ class RLNAS(object): ...@@ -75,6 +75,7 @@ class RLNAS(object):
self.range_tables = self._search_space.range_table() self.range_tables = self._search_space.range_table()
self.save_controller = save_controller self.save_controller = save_controller
self.load_controller = load_controller self.load_controller = load_controller
self._is_server = is_server
if key.upper() in ['DDPG']: if key.upper() in ['DDPG']:
try: try:
......
...@@ -22,7 +22,6 @@ import time ...@@ -22,7 +22,6 @@ import time
import paddle.fluid as fluid import paddle.fluid as fluid
from ..common import SAController from ..common import SAController
from ..common import get_logger from ..common import get_logger
from ..analysis import flops
from ..common import ControllerServer from ..common import ControllerServer
from ..common import ControllerClient from ..common import ControllerClient
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册