fix inferface issue

485f7033 · wuzewu · GitHub · a89ca64d · 485f7033
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

pdseg/train.py pdseg/train.py +1 -1

未找到文件。
--- a/pdseg/train.py
+++ b/pdseg/train.py
@@ -462,7 +462,7 @@ def main(args):
    cfg.TRAINER_ID = int(os.getenv("PADDLE_TRAINER_ID", 0))
    cfg.NUM_TRAINERS = int(os.environ.get('PADDLE_TRAINERS_NUM', 1))
-    cfg.check_and_infer(reset_dataset=True)
+    cfg.check_and_infer()
    print_info(pprint.pformat(cfg))
    train(cfg)