Commit f9725a91 by Wenqi Li

Merge branch 'patch-4' into 'dev'

Update segmentation_application.py

See merge request !148
parents 87257623 a3ad0a4d
Pipeline #10047 passed with stages
in 3 minutes 2 seconds
......@@ -290,10 +290,10 @@ class SegmentationApplication(BaseApplication):
gradients_collector.add_to_collection([grads])
# collecting output variables
outputs_collector.add_to_collection(
var=data_loss, name='dice_loss',
var=data_loss, name='loss',
average_over_devices=False, collection=CONSOLE)
outputs_collector.add_to_collection(
var=data_loss, name='dice_loss',
var=data_loss, name='loss',
average_over_devices=True, summary_type='scalar',
collection=TF_SUMMARIES)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment