funasr/frontends/s3prl.py
@@ -102,9 +102,9 @@ Output - sequence of tiled representations shape: (batch_size, seq_len * factor, feature_dim) """ assert ( len(feature.shape) == 3 ), "Input argument `feature` has invalid shape: {}".format(feature.shape) assert len(feature.shape) == 3, "Input argument `feature` has invalid shape: {}".format( feature.shape ) tiled_feature = feature.repeat(1, 1, self.args.tile_factor) tiled_feature = tiled_feature.reshape( feature.size(0), feature.size(1) * self.args.tile_factor, feature.size(2)
