Update NAS Demo (PaddlePaddle#1743)

lizexu123 · May 6, 2023 · 673c158 · 673c158
1 parent 18eebf8
commit 673c158
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/demo/nas/sa_nas_mobilenetv2.py b/demo/nas/sa_nas_mobilenetv2.py
@@ -140,12 +140,12 @@ def search_mobilenetv2(config, args, image_size, is_server=True):
                 fetches = [avg_cost.name]
                 s_time = time.time()
                 outs = exe.run(
-                    train_compiled_program, feed=data, fetch_list=fetches)[0]
+                    train_compiled_program, feed=data[0], fetch_list=fetches)
                 batch_time = time.time() - s_time
                 if batch_id % 10 == 0:
                     _logger.info(
                         'TRAIN: steps: {}, epoch: {}, batch: {}, cost: {}, batch_time: {}ms'.
-                        format(step, epoch_id, batch_id, outs[0], batch_time))
+                        format(step, epoch_id, batch_id, outs, batch_time))
 
         reward = []
         for batch_id, data in enumerate(test_loader()):