From 5b972ed9b03d7d9234aad21075e36b7192b1eb79 Mon Sep 17 00:00:00 2001 From: Justin Date: Thu, 23 Mar 2023 20:23:14 +0000 Subject: [PATCH] smaller thing --- nerfstudio/fields/lerf_field.py | 7 ++++--- nerfstudio/models/lerf.py | 2 +- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/nerfstudio/fields/lerf_field.py b/nerfstudio/fields/lerf_field.py index 358178a42..242e472b9 100644 --- a/nerfstudio/fields/lerf_field.py +++ b/nerfstudio/fields/lerf_field.py @@ -28,7 +28,8 @@ class LERFField(Field): self.spatial_distortion = spatial_distortion self.clip_encs = torch.nn.ModuleList( - [LERFField._get_encoding(16, 128, 16, indim=3), LERFField._get_encoding(128, 512, 16, indim=3)] + # [LERFField._get_encoding(16, 512, 16, indim=3)] + [LERFField._get_encoding(16, 128, 12, indim=3), LERFField._get_encoding(128, 512, 12, indim=3)] ) tot_out_dims = sum([e.n_output_dims for e in self.clip_encs]) @@ -40,7 +41,7 @@ class LERFField(Field): "activation": "ReLU", "output_activation": "None", "n_neurons": 256, - "n_hidden_layers": 3, + "n_hidden_layers": 4, }, ) @@ -65,7 +66,7 @@ class LERFField(Field): "otype": "HashGrid", "n_levels": levels, "n_features_per_level": 8, - "log2_hashmap_size": 21, + "log2_hashmap_size": 19, "base_resolution": start_res, "per_level_scale": growth, }, diff --git a/nerfstudio/models/lerf.py b/nerfstudio/models/lerf.py index ef83f352f..30d197a37 100644 --- a/nerfstudio/models/lerf.py +++ b/nerfstudio/models/lerf.py @@ -61,7 +61,7 @@ class LERFModel(NerfactoModel): clip_output = self.lerf_field.get_output_from_hashgrid( ray_samples, hashgrid_field, - torch.full(scales_shape, s, device=weights.device, dtype=hashgrid_field.dtype), + torch.full(scales_shape, scale, device=weights.device, dtype=hashgrid_field.dtype), ) clip_output = self.renderer_clip(embeds=clip_output, weights=weights.detach())