GitHub repository: nxdens/linearInit
Primary language: Python
Created: 2023-10-19T17:40:41Z
Description: Wraps the pytorch nn.linear layer with various initialization schemes based on the non-linear activation used