diff --git a/models/__init__.py b/models/__init__.py index f090d2c..a5b7cb3 100644 --- a/models/__init__.py +++ b/models/__init__.py @@ -54,7 +54,7 @@ def __init__(self, main_net: nn.Module, appearance_net=None, near=2, far=7, white_bg=True, use_alpha=True, min_alpha_requirement=1e-4, normalize_coords=True, - density_scale=1, # 25, + density_scale=25, density_activation=None, appearance_activation=None): super().__init__() """