Ali Mohsin
feat: Add virtual try-on system components including DensePose, SMPL, and pix2pixHD models, rendering, and utilities.
5db43ff
| # Copyright (c) Facebook, Inc. and its affiliates. | |
| from torch import nn | |
| def initialize_module_params(module: nn.Module) -> None: | |
| for name, param in module.named_parameters(): | |
| if "bias" in name: | |
| nn.init.constant_(param, 0) | |
| elif "weight" in name: | |
| nn.init.kaiming_normal_(param, mode="fan_out", nonlinearity="relu") | |