Yo aplique esto y me corrio en la instacicion.
def weights_init_normal(m):
if type(m) == nn.Conv2d:
torch.nn.init.normal_(m.weight.data, 0.0, 0.02)
elif type(m) == nn.BatchNorm2d:
torch.nn.init.normal_(m.weight.data, 1.0, 0.02)
torch.nn.init.constant(m.bias, 0.0)
Estaba dentro de los ejemplos de…
help(nn.ReflectionPad2d)
🤓
¿Quieres ver más aportes, preguntas y respuestas de la comunidad? Crea una cuenta o inicia sesión.