From 867df87a13915c6d2d9dc77e62b94fbf6df8d3e2 Mon Sep 17 00:00:00 2001 From: hxcai Date: Mon, 29 Mar 2021 19:37:40 +0800 Subject: [PATCH] fix assembler bug --- vedadet/assembler/trainval.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vedadet/assembler/trainval.py b/vedadet/assembler/trainval.py index ab4831e..5f4df7c 100644 --- a/vedadet/assembler/trainval.py +++ b/vedadet/assembler/trainval.py @@ -22,7 +22,8 @@ def trainval(cfg, distributed, logger): dataset, cfg.data.samples_per_gpu, cfg.data.workers_per_gpu, - dist=distributed) + dist=distributed, + seed=cfg.get('seed', None)) engine = build_engine(cfg.train_engine) if distributed: