提交 f54729cb 编写于 作者: Q qjing666
...@@ -18,4 +18,3 @@ from . import common ...@@ -18,4 +18,3 @@ from . import common
from . import core from . import core
from . import dataset from . import dataset
from . import reader from . import reader
...@@ -11,5 +11,3 @@ ...@@ -11,5 +11,3 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
...@@ -146,4 +146,3 @@ class JobGenerator(object): ...@@ -146,4 +146,3 @@ class JobGenerator(object):
local_job.set_target_names(self._target_names) local_job.set_target_names(self._target_names)
local_job.set_strategy(fl_strategy) local_job.set_strategy(fl_strategy)
local_job.save(output) local_job.save(output)
...@@ -273,5 +273,3 @@ class SecAggStrategy(FedAvgStrategy): ...@@ -273,5 +273,3 @@ class SecAggStrategy(FedAvgStrategy):
@param_name_list.setter @param_name_list.setter
def param_name_list(self, s): def param_name_list(self, s):
self._param_name_list = s self._param_name_list = s
...@@ -16,7 +16,7 @@ import logging ...@@ -16,7 +16,7 @@ import logging
from paddle_fl.core.scheduler.agent_master import FLWorkerAgent from paddle_fl.core.scheduler.agent_master import FLWorkerAgent
import numpy import numpy
import hmac import hmac
from .diffiehellman.diffiehellman import DiffieHellman #from .diffiehellman.diffiehellman import DiffieHellman
class FLTrainerFactory(object): class FLTrainerFactory(object):
def __init__(self): def __init__(self):
...@@ -149,6 +149,9 @@ class FedAvgTrainer(FLTrainer): ...@@ -149,6 +149,9 @@ class FedAvgTrainer(FLTrainer):
return loss return loss
class SecAggTrainer(FLTrainer): class SecAggTrainer(FLTrainer):
def __init__(self): def __init__(self):
super(SecAggTrainer, self).__init__() super(SecAggTrainer, self).__init__()
...@@ -244,4 +247,3 @@ class SecAggTrainer(FLTrainer): ...@@ -244,4 +247,3 @@ class SecAggTrainer(FLTrainer):
def stop(self): def stop(self):
return False return False
...@@ -68,5 +68,3 @@ while not trainer.stop(): ...@@ -68,5 +68,3 @@ while not trainer.stop():
save_dir = (output_folder + "/epoch_%d") % epoch_id save_dir = (output_folder + "/epoch_%d") % epoch_id
trainer.save_inference_program(output_folder) trainer.save_inference_program(output_folder)
...@@ -71,4 +71,3 @@ setup( ...@@ -71,4 +71,3 @@ setup(
], ],
license='Apache 2.0', license='Apache 2.0',
keywords=('paddle_fl paddlepaddle multi-task transfer distributed-training')) keywords=('paddle_fl paddlepaddle multi-task transfer distributed-training'))
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册