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

classification model was updated

parent 9c6039c5
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 3 additions and 3 deletions
dgcnn/dgcnn_train_pl.py
+ 3
− 3
View file @ 84d50a5b
......@@ -138,7 +138,7 @@ shapenet_data_train = ShapenetDataDgcnn(
return_cls_label=True,
small_data=config['data']['small_data'],
small_data_size=config['data']['small_data_size'],
just_four_classes=config['data']['just_four_classes'],
num_classes=config['data']['just_four_classes'],
split='train',
norm=config['data']['norm'],
augmnetation=config['data']['augmentation']
......@@ -151,7 +151,7 @@ shapenet_data_val = ShapenetDataDgcnn(
return_cls_label=True,
small_data=config['data']['small_data'],
small_data_size=config['data']['small_data_size'],
just_four_classes=config['data']['just_four_classes'],
num_classes=config['data']['just_four_classes'],
split='test',
norm=config['data']['norm']
)
......@@ -163,7 +163,7 @@ shapenet_data_test = ShapenetDataDgcnn(
return_cls_label=True,
small_data=config['data']['small_data'],
small_data_size=config['data']['small_data_size'],
just_four_classes=config['data']['just_four_classes'],
num_classes=config['data']['just_four_classes'],
split='test',
norm=config['data']['norm']
)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment