-
Notifications
You must be signed in to change notification settings - Fork 28
Open
Labels
testwrite or fix testswrite or fix tests
Description
GaborNet/GaborNet/GaborLayer.py
Lines 52 to 60 in a104371
self.theta = Parameter( | |
(math.pi / 8) | |
* torch.randint(0, 8, (out_channels, in_channels)).type(torch.Tensor), | |
requires_grad=True, | |
) | |
self.sigma = Parameter(math.pi / self.freq, requires_grad=True) | |
self.psi = Parameter( | |
math.pi * torch.rand(out_channels, in_channels), requires_grad=True | |
) |
Metadata
Metadata
Assignees
Labels
testwrite or fix testswrite or fix tests