File size: 816 Bytes
81df686 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
# Copyright (c) ByteDance, Inc. and its affiliates.
# Copyright (c) Chutong Meng
#
# This source code is licensed under the CC BY-NC license found in the
# LICENSE file in the root directory of this source tree.
# Based on AudioDec (https://github.com/facebookresearch/AudioDec)
import torch.nn as nn
from repcodec.layers.conv_layer import Conv1d
class Projector(nn.Module):
def __init__(
self,
input_channels: int,
code_dim: int,
kernel_size=3,
stride=1,
bias=False
):
super().__init__()
self.project = Conv1d(
input_channels,
code_dim,
kernel_size=kernel_size,
stride=stride,
bias=bias
)
def forward(self, x):
return self.project(x)
|