def print_shape(self, input_shape): print("super-resolution {") SequentialMaker.print_shape(self.seq_, torch.FloatTensor(np.zeros(input_shape)), intent_size=2) print("}")
def print_shape(self, input_shape): print("text-encode {") SequentialMaker.print_shape(self.seq_, torch.LongTensor(np.zeros(input_shape)), intent_size=2) print("}")
def print_shape(self, input_shape): print("audio-decoder {") SequentialMaker.print_shape(self.seq_, torch.FloatTensor(np.zeros(input_shape)), intent_size=2) print("}")