Skip to content
Snippets Groups Projects
Commit 7a272bc0 authored by Maciej Wielgosz's avatar Maciej Wielgosz
Browse files

train forest updated

parent f23e2ff4
No related branches found
No related tags found
No related merge requests found
...@@ -55,8 +55,8 @@ def main(args): ...@@ -55,8 +55,8 @@ def main(args):
# use pretty print to print the config # use pretty print to print the config
train_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/validation_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) 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) trainDataLoader = torch.utils.data.DataLoader(TRAIN_DATASET, batch_size=args.batch_size, shuffle=True, num_workers=10, drop_last=True)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment