def output_types(self): """Returns definitions of module output types """ return { "augmented_spec": NeuralType(('B', 'D', 'T'), SpectrogramType()) }
def output_types(self): return OrderedDict( {"spec_recon": NeuralType(('B', 'T', 'D'), SpectrogramType())})
def output_types(self) -> Optional[Dict[str, NeuralType]]: return { "spectrograms": NeuralType(('B', 'D', 'T'), SpectrogramType()), "spec_masks": NeuralType(('B', 'D', 'T'), SpectrogramType()), "outputs": NeuralType(('B', 'T', 'D'), VoidType()), }
def input_types(self): """Returns definitions of module input types """ return {"input_spec": NeuralType(('B', 'D', 'T'), SpectrogramType())}