-
Benjamin Vandersmissen authored
Added a fix to ensure that the pruning masks are on the same device as the weights Cleanup of the main loop Added a way to select the initialization of layers, including the sparse-aware initialization from [Evci2022]
87c962a1