espnet.nets.pytorch_backend.transformer.initializer.initialize
Less than 1 minute
espnet.nets.pytorch_backend.transformer.initializer.initialize
espnet.nets.pytorch_backend.transformer.initializer.initialize(model, init_type='pytorch')
Initialize Transformer module.
- Parameters:
- model (torch.nn.Module) – transformer instance
- init_type (str) – initialization type