diff --git a/paddle/fluid/train/custom_trainer/feed/scripts/join.py b/paddle/fluid/train/custom_trainer/feed/scripts/join.py index d3aeceb45062364745f3d316a35db3829f3f9efe..42279410da99e7049fd12e0b7bb1e46fe40afe62 100644 --- a/paddle/fluid/train/custom_trainer/feed/scripts/join.py +++ b/paddle/fluid/train/custom_trainer/feed/scripts/join.py @@ -50,7 +50,14 @@ def inference(): { "name": "epoch_auc", "class": "AucMonitor", "target": ctr_output, "compute_interval": 600 }, { "name": "day_auc", "class": "AucMonitor", "target": ctr_output, "compute_interval": 86400 } ] - return {'accessors': accessors, 'monitors': monitors, 'sparses': [sparse_cvm], 'inputs': [cvm_input], 'outputs': [ctr_output]} + + return { + 'accessors': accessors, + 'monitors': monitors, + 'sparses': [sparse_cvm], + 'inputs': [cvm_input], + 'outputs': [ctr_output] + } def loss_function(ctr_output): """ diff --git a/paddle/fluid/train/custom_trainer/feed/scripts/update.py b/paddle/fluid/train/custom_trainer/feed/scripts/update.py index ec77ba16e86b6055c314653afdac3eec595c88f7..c89ada5de133dd35e466f59d7ebf935d679b1302 100644 --- a/paddle/fluid/train/custom_trainer/feed/scripts/update.py +++ b/paddle/fluid/train/custom_trainer/feed/scripts/update.py @@ -43,7 +43,13 @@ def inference(): { "name": "day_auc", "class": "AucMonitor", "target": ctr_output, "compute_interval": 86400 } ] - return {'accessors': accessors, 'monitors': monitors, 'sparses': [sparse_cvm], 'inputs': [cvm_input], 'outputs': [ctr_output]} + return { + 'accessors': accessors, + 'monitors': monitors, + 'sparses': [sparse_cvm], + 'inputs': [cvm_input], + 'outputs': [ctr_output] + } def loss_function(ctr_output): """