Skip to content

Commit 5641739

Browse files
committed
Add num_eval when dataste is not specified
1 parent 49b322d commit 5641739

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/loader.py

+2
Original file line numberDiff line numberDiff line change
@@ -294,6 +294,8 @@ def load_worker(local_rank, cfgs, gpus_per_node, run_name, hdf5_path):
294294
else:
295295
if eval_dataloader is not None:
296296
num_eval[cfgs.RUN.ref_dataset] = len(eval_dataloader.dataset)
297+
else:
298+
num_eval["train"], num_eval["valid"], num_eval["test"] = 50000, 50000, 50000
297299

298300
if len(cfgs.RUN.eval_metrics) or cfgs.RUN.intra_class_fid:
299301
eval_model = pp.LoadEvalModel(eval_backbone=cfgs.RUN.eval_backbone,

0 commit comments

Comments
 (0)