(self, input_dim=3, degree=4)
| 54 | |
| 55 | class FreqEncoder(nn.Module): |
| 56 | def __init__(self, input_dim=3, degree=4): |
| 57 | super().__init__() |
| 58 | |
| 59 | self.input_dim = input_dim |
| 60 | self.degree = degree |
| 61 | self.output_dim = input_dim + input_dim * 2 * degree |
| 62 | |
| 63 | def __repr__(self): |
| 64 | return f"FreqEncoder: input_dim={self.input_dim} degree={self.degree} output_dim={self.output_dim}" |
nothing calls this directly
no outgoing calls
no test coverage detected