Skip to content

Add test cases to check that in train mode params change #20

@iKintosh

Description

@iKintosh

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

No one assigned

    Labels

    testwrite or fix tests

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions