diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/fused_pull_weight_kernel.h b/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/fused_pull_weight_kernel.h
index 641fa10ff2b..4548d728803 100644
--- a/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/fused_pull_weight_kernel.h
+++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/fused_pull_weight_kernel.h
@@ -63,6 +63,7 @@ class FusedPullWeightKernel : public CPUKernel {
       fl_iteration_ = 1;
     }
 
+    MS_LOG(INFO) << "Start pulling weight for federated learning iteration " << fl_iteration_;
     if (!BuildPullWeightReq(fbb)) {
       MS_LOG(EXCEPTION) << "Building request for FusedPullWeight failed.";
       return false;
diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/fused_push_weight_kernel.h b/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/fused_push_weight_kernel.h
index 1caf3489499..e9590764c89 100644
--- a/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/fused_push_weight_kernel.h
+++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/fused_push_weight_kernel.h
@@ -61,6 +61,7 @@ class FusedPushWeightKernel : public CPUKernel {
       fl_iteration_ = 1;
     }
 
+    MS_LOG(INFO) << "Start pushing weight for federated learning iteration " << fl_iteration_;
     if (!BuildPushWeightReq(fbb, inputs)) {
       MS_LOG(EXCEPTION) << "Building request for FusedPushWeight failed.";
       return false;
diff --git a/mindspore/parallel/_ps_context.py b/mindspore/parallel/_ps_context.py
index 6a9f7897a59..6ef057e1708 100644
--- a/mindspore/parallel/_ps_context.py
+++ b/mindspore/parallel/_ps_context.py
@@ -146,6 +146,7 @@ def _set_ps_context(**kwargs):
     for key, value in kwargs.items():
         if key not in _set_ps_context_func_map:
             raise ValueError("Set PS context keyword %s is not recognized!" % key)
+        _check_value(key, value)
         set_func = _set_ps_context_func_map[key]
         set_func(value)