From 736e72914a5403a2f6d4a4bffa9154864402b327 Mon Sep 17 00:00:00 2001 From: Matt Tancik Date: Thu, 2 Jun 2022 12:10:42 -0700 Subject: [PATCH] Fix registered value. --- pyrad/nerf/field_modules/encoding.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyrad/nerf/field_modules/encoding.py b/pyrad/nerf/field_modules/encoding.py index 3b65039b1..75c058ecf 100644 --- a/pyrad/nerf/field_modules/encoding.py +++ b/pyrad/nerf/field_modules/encoding.py @@ -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