From 1bf1feeeab4c0422531af04be0de54de53512e17 Mon Sep 17 00:00:00 2001
From: HuguesTHOMAS <huguesthomas218@gmail.com>
Date: Tue, 31 Mar 2020 15:47:53 -0400
Subject: [PATCH] Initial commit

---
 train_ModelNet40.py | 2 +-
 train_S3DIS.py      | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/train_ModelNet40.py b/train_ModelNet40.py
index 33e1f20..40af46c 100644
--- a/train_ModelNet40.py
+++ b/train_ModelNet40.py
@@ -119,7 +119,7 @@ class Modelnet40Config(Config):
     # 'permissive' only constrains offsets inside the deform radius
     # 'fitting' helps deformed kernels to adapt to the geometry by penalizing distance to input points
     offsets_loss = 'fitting'
-    offsets_decay = 1.0
+    offsets_decay = 0.1
 
     #####################
     # Training parameters
diff --git a/train_S3DIS.py b/train_S3DIS.py
index 210f9d0..9c50ac2 100644
--- a/train_S3DIS.py
+++ b/train_S3DIS.py
@@ -74,10 +74,10 @@ class S3DISConfig(Config):
                     'resnetb',
                     'resnetb_strided',
                     'resnetb',
-                    'resnetb_deformable_strided',
-                    'resnetb_deformable',
-                    'resnetb_deformable_strided',
-                    'resnetb_deformable',
+                    'resnetb_strided',
+                    'resnetb',
+                    'resnetb_strided',
+                    'resnetb',
                     'nearest_upsample',
                     'unary',
                     'nearest_upsample',