(self,emb_dim,fea_dim)
| 39 | |
| 40 | class VideoEncoder(nn.Module): |
| 41 | def __init__(self,emb_dim,fea_dim): |
| 42 | super(VideoEncoder, self).__init__() |
| 43 | |
| 44 | self.emb_dim = emb_dim |
| 45 | self.linear1 = torch.nn.Linear(self.emb_dim, self.emb_dim, bias=False) |
| 46 | self.linear2 = nn.Sequential(torch.nn.Linear(self.emb_dim, fea_dim),torch.nn.ReLU()) |
| 47 | |
| 48 | def forward(self, input_thumb, input_L): |
| 49 | input_ALL = torch.cat((input_L, input_thumb),1) #(bs,len+1,4096) |