Skip to content

Commit

Permalink
initialize loc-spec params from gaussian
Browse files Browse the repository at this point in the history
  • Loading branch information
henryaddison committed Mar 4, 2024
1 parent 03ebfec commit cec39d3
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ class LocationParams(torch.nn.Module):
def __init__(self, n_channels, size) -> None:
super().__init__()

self.params = torch.nn.Parameter(torch.zeros(n_channels, size, size))
self.params = torch.nn.Parameter(torch.randn(n_channels, size, size))

def forward(self, cond):
batch_size = cond.shape[0]
Expand Down

0 comments on commit cec39d3

Please sign in to comment.