taesungp/pytorch-kmeans-init

Weight initialization of deep networks using k-means clustering

Python