From 84d50a5b84b070e8e7ad654b9fe2e7e779195472 Mon Sep 17 00:00:00 2001 From: Maciej Wielgosz <maciej.wielgosz@nibio.no> Date: Fri, 21 Apr 2023 14:39:50 +0200 Subject: [PATCH] classification model was updated --- dgcnn/dgcnn_train_pl.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/dgcnn/dgcnn_train_pl.py b/dgcnn/dgcnn_train_pl.py index c76f4c4..e49a943 100644 --- a/dgcnn/dgcnn_train_pl.py +++ b/dgcnn/dgcnn_train_pl.py @@ -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'] ) -- GitLab