diff --git a/train_ModelNet40.py b/train_ModelNet40.py index 3f5ef08..feaa048 100644 --- a/train_ModelNet40.py +++ b/train_ModelNet40.py @@ -126,7 +126,7 @@ class Modelnet40Config(Config): deform_fitting_mode = 'point2point' deform_fitting_power = 1.0 # Multiplier for the fitting/repulsive loss deform_lr_factor = 0.1 # Multiplier for learning rate applied to the deformations - repulse_extent = 0.8 # Distance of repulsion for deformed kernel points + repulse_extent = 1.2 # Distance of repulsion for deformed kernel points ##################### # Training parameters diff --git a/train_SemanticKitti.py b/train_SemanticKitti.py index cd98834..b474bd8 100644 --- a/train_SemanticKitti.py +++ b/train_SemanticKitti.py @@ -148,7 +148,7 @@ class SemanticKittiConfig(Config): deform_fitting_mode = 'point2point' deform_fitting_power = 1.0 # Multiplier for the fitting/repulsive loss deform_lr_factor = 0.1 # Multiplier for learning rate applied to the deformations - repulse_extent = 0.8 # Distance of repulsion for deformed kernel points + repulse_extent = 1.2 # Distance of repulsion for deformed kernel points ##################### # Training parameters