From df80c2f07b05a87cdc0a9b346dadf389266f9b5e Mon Sep 17 00:00:00 2001 From: shippingwang Date: Mon, 29 Jun 2020 07:07:03 +0000 Subject: [PATCH] fix bug --- tools/program.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tools/program.py b/tools/program.py index a61225cb..71e9200e 100644 --- a/tools/program.py +++ b/tools/program.py @@ -151,7 +151,6 @@ def create_loss(out, loss = JSDivLoss(class_dim=classes_num, epsilon=epsilon) return loss(out[1], out[0]) - print("++++++", use_mix) if use_mix: loss = MixCELoss(class_dim=classes_num, epsilon=epsilon) feed_y_a = feeds['feed_y_a'] @@ -341,7 +340,7 @@ def build(config, main_prog, startup_prog, is_train=True): use_mix = config.get('use_mix') and is_train use_dali = config.get('use_dali') use_distillation = config.get('use_distillation') - feeds = create_feeds(config.image_shape, use_mix=use_mix, use_dali) + feeds = create_feeds(config.image_shape, use_mix, use_dali) if use_dali and use_mix: import dali -- GitLab