File size: 362 Bytes
5d9b308 | 1 2 3 4 5 6 7 8 9 10 11 12 13 | # SPDX-License-Identifier: MIT
# Adapted from https://github.com/jik876/hifi-gan under the MIT license.
def init_weights(m, mean=0.0, std=0.01):
classname = m.__class__.__name__
if classname.find("Conv") != -1:
m.weight.data.normal_(mean, std)
def get_padding(kernel_size, dilation=1):
return int((kernel_size * dilation - dilation) / 2)
|