diff --git a/paddle/fluid/train/custom_trainer/feed/scripts/join.py b/paddle/fluid/train/custom_trainer/feed/scripts/join.py index 2eb3da7f7eeda633ebc69540d134359ae00138a6..f0317adf7ad49ee13a13b56384893bb5e5747bc4 100644 --- a/paddle/fluid/train/custom_trainer/feed/scripts/join.py +++ b/paddle/fluid/train/custom_trainer/feed/scripts/join.py @@ -55,7 +55,8 @@ def inference(): { "class": "AbacusSparseJoinAccessor", "input": "sparses", "table_id": 0, "need_gradient": False}, { "class": "DenseInputAccessor", "input": "vars", "table_id": 1, "need_gradient": True, "async_pull": True}, { "class": "DenseInputAccessor", "input": "sums", "table_id": 2, "need_gradient": True, "async_pull": True}, - { "class": "WeightsAdjustAccessor", "input": "ins_weight", "slot_id": 6002, "adjw_ratio": 20, "adjw_threshold": 1000 }, + { "class": "WeightsAdjustAccessor", "input": "ins_weight", + "slot_id": 6002, "adjw_ratio": 20, "adjw_threshold": 1000 }, { "class": "LabelInputAccessor", "input": "labels"} ] diff --git a/paddle/fluid/train/custom_trainer/feed/scripts/update.py b/paddle/fluid/train/custom_trainer/feed/scripts/update.py index 7fe76be0385b463ce74d58a333ef2693d7b16d72..aa5c2f3f0f020146d461d0967b178413891c648a 100644 --- a/paddle/fluid/train/custom_trainer/feed/scripts/update.py +++ b/paddle/fluid/train/custom_trainer/feed/scripts/update.py @@ -51,7 +51,8 @@ def inference(): accessors = [ { "class": "AbacusSparseUpdateAccessor", "input": "sparses", "table_id": 0, "need_gradient": True}, { "class": "DenseInputAccessor", "input": "vars", "table_id": 5, "need_gradient": True, "async_pull": True}, - { "class": "WeightsAdjustAccessor", "input": "ins_weight", "slot_id": 6002, "adjw_ratio": 20, "adjw_threshold": 1000 }, + { "class": "WeightsAdjustAccessor", "input": "ins_weight", + "slot_id": 6002, "adjw_ratio": 20, "adjw_threshold": 1000}, { "class": "LabelInputAccessor", "input": "labels"} ] monitors = [