Spaces:
Build error
Build error
File size: 240 Bytes
e487255 |
1 2 3 4 5 6 7 8 9 |
import torch
from torch import nn
def get_NormalInitializer(std):
def initializer(m):
if isinstance(m, nn.Linear):
nn.init.normal_(m.weight, 0, std)
nn.init.normal_(m.bias, 0, std)
return initializer |