From 54633e453192250b4ce1888bb28a7eb9acf577a2 Mon Sep 17 00:00:00 2001 From: Andrew White Date: Mon, 27 Jul 2026 20:47:58 -0500 Subject: [PATCH] fix: test dataset log message prints train_dataset --- dust3r/training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dust3r/training.py b/dust3r/training.py index 53af976..8dbb397 100644 --- a/dust3r/training.py +++ b/dust3r/training.py @@ -119,7 +119,7 @@ def train(args): print('Building train dataset {:s}'.format(args.train_dataset)) # dataset and loader data_loader_train = build_dataset(args.train_dataset, args.batch_size, args.num_workers, test=False) - print('Building test dataset {:s}'.format(args.train_dataset)) + print('Building test dataset {:s}'.format(args.test_dataset)) data_loader_test = {dataset.split('(')[0]: build_dataset(dataset, args.batch_size, args.num_workers, test=True) for dataset in args.test_dataset.split('+')}