mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-11 21:41:18 +00:00
arensc
14 lines
376 B
Python
14 lines
376 B
Python
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 apply_weight_norm(m):
|
|
classname = m.__class__.__name__
|
|
if classname.find("Conv") != -1:
|
|
weight_norm(m)
|
|
|
|
|
|
def get_padding(kernel_size, dilation=1):
|
|
return int((kernel_size*dilation - dilation)/2) |