Commit 7f2f8238 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

add convtranspose1D

parent a8d9f3ed
......@@ -440,6 +440,11 @@ class Xtractor(torch.nn.Module):
dilation=cfg["segmental"][k]["dilation"])))
input_size = cfg["segmental"][k]["output_channels"]
elif k.startswith("ctrans"):
segmental_layers.append((k, torch.nn.ConvTranspose1d(input_size,
cfg["segmental"][k]["output_channels"],
kernel_size=cfg["segmental"][k]["kernel_size"],
dilation=cfg["segmental"][k]["dilation"])))
elif k.startswith("activation"):
segmental_layers.append((k, self.activation))
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment