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