From 4caaa288f6f3aebe4002edc9b9ec14b186a92d73 Mon Sep 17 00:00:00 2001 From: dengkaipeng Date: Sun, 5 Apr 2020 12:55:54 +0000 Subject: [PATCH] unset feed_list --- yolov3/main.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/yolov3/main.py b/yolov3/main.py index c659b5f..18c24d1 100644 --- a/yolov3/main.py +++ b/yolov3/main.py @@ -91,8 +91,6 @@ def main(): loader = DataLoader(dataset, batch_sampler=batch_sampler, places=device, - feed_list=[i.forward() for i in inputs + labels] \ - if not FLAGS.dynamic else None, num_workers=FLAGS.num_workers, return_list=True, collate_fn=train_collate_fn) @@ -116,8 +114,6 @@ def main(): loader = DataLoader(dataset, batch_sampler=batch_sampler, places=device, - feed_list=[i.forward() for i in inputs + labels] \ - if not FLAGS.dynamic else None, num_workers=FLAGS.num_workers, return_list=True, collate_fn=eval_collate_fn) -- GitLab