Time2Vec-PyTorch

Time2Vec-PyTorch copied to clipboard

why not write it using `nn.Linear`?

1

Here is a simple version. ```Python import torch import torch.nn as nn class Time2vec(nn.Module): def __init__(self, c_in, c_out, activation="cos"): super().__init__() self.wnbn = nn.Linear(c_in, c_out - 1, bias=True) self.w0b0 = nn.Linear(c_in,...

crazydogen

Why Linear output is 2 for Model

2

Can you please explain why we have got 2 dim for fc output in Model.py. ?

jaideep11061982