提交 94299aa8 编写于 作者: K Kaipeng Deng 提交者: GitHub

fix multi-process reader not exist in resume (#3323)

上级 830297cb
...@@ -103,10 +103,6 @@ def main(): ...@@ -103,10 +103,6 @@ def main():
optimizer = optim_builder(lr) optimizer = optim_builder(lr)
optimizer.minimize(loss) optimizer.minimize(loss)
train_reader = create_reader(train_feed, cfg.max_iters * devices_num,
FLAGS.dataset_dir)
train_pyreader.decorate_sample_list_generator(train_reader, place)
# parse train fetches # parse train fetches
train_keys, train_values, _ = parse_fetches(train_fetches) train_keys, train_values, _ = parse_fetches(train_fetches)
train_values.append(lr) train_values.append(lr)
...@@ -163,6 +159,13 @@ def main(): ...@@ -163,6 +159,13 @@ def main():
elif cfg.pretrain_weights: elif cfg.pretrain_weights:
checkpoint.load_pretrain(exe, train_prog, cfg.pretrain_weights) checkpoint.load_pretrain(exe, train_prog, cfg.pretrain_weights)
train_reader = create_reader(
train_feed,
(cfg.max_iters - start_iter) * devices_num,
FLAGS.dataset_dir)
train_pyreader.decorate_sample_list_generator(train_reader, place)
# whether output bbox is normalized in model output layer # whether output bbox is normalized in model output layer
is_bbox_normalized = False is_bbox_normalized = False
if hasattr(model, 'is_bbox_normalized') and \ if hasattr(model, 'is_bbox_normalized') and \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册