From 7a272bc00ef724c8b78f8d01d2fc52b3861dcc8d Mon Sep 17 00:00:00 2001
From: Maciej Wielgosz <maciej.wielgosz@nibio.no>
Date: Thu, 9 Mar 2023 11:17:59 +0100
Subject: [PATCH] train forest updated
---
train_partseg_forest.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/train_partseg_forest.py b/train_partseg_forest.py
index f3ba5a7..a2c83a1 100644
--- a/train_partseg_forest.py
+++ b/train_partseg_forest.py
@@ -55,8 +55,8 @@ def main(args):
# use pretty print to print the config
- train_dataset = hydra.utils.to_absolute_path('data/forest_txt/validation_txt/')
- test_dataset = hydra.utils.to_absolute_path('data/forest_txt/validation_txt/')
+ train_dataset = hydra.utils.to_absolute_path('data/forest_txt/train_txt/')
+ test_dataset = hydra.utils.to_absolute_path('data/forest_txt/test_txt/')
TRAIN_DATASET = Dataset(root=train_dataset, npoints=args.num_point, normal_channel=args.normal)
trainDataLoader = torch.utils.data.DataLoader(TRAIN_DATASET, batch_size=args.batch_size, shuffle=True, num_workers=10, drop_last=True)
--
GitLab