tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-14 11:54:01 +08:00
Code Activity

Merge pull request #10 from plenoptix/tancik/fix_rff_state_save

Browse Source 
Fix registered value.
This commit is contained in:
Matthew Tancik authored and GitHub committed 2022-06-02 13:38:50 -07:00
2 parents ebc64a13bf 736e72914a
commit 1a26e845b0
1 file changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files
pyrad/nerf/field_modules/encoding.py
+2 -2
View File
@@ -138,8 +138,8 @@ class RFFEncoding(Encoding):
if not scale > 0:
raise ValueError("RFF encoding scale should be greater than zero")
self.scale = scale
self.b_matrix = torch.normal(mean=0, std=self.scale, size=(self.in_dim, self.num_frequencies))
self.register_buffer(name="rff_scalings", tensor=self.b_matrix)
b_matrix = torch.normal(mean=0, std=self.scale, size=(self.in_dim, self.num_frequencies))
self.register_buffer(name="b_matrix", tensor=b_matrix)
def get_out_dim(self) -> int:
return self.num_frequencies * 2