diff --git a/fluid/PaddleCV/image_classification/train.py b/fluid/PaddleCV/image_classification/train.py index e15357c79540fcab88ef0780652f4051d01308bd..7fdf0a7e78517093382142c5776ba73add582c35 100644 --- a/fluid/PaddleCV/image_classification/train.py +++ b/fluid/PaddleCV/image_classification/train.py @@ -313,10 +313,6 @@ def train(args): batch_id = 0 try: while True: - if pass_id == 0 and batch_id == 5: - profiler.start_profiler("GPU") - elif pass_id == 0 and batch_id == 10: - profiler.stop_profiler("total", "profile") t1 = time.time() loss, acc1, acc5, lr = train_exe.run( fetch_list=train_fetch_list)