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',