pytorch-kmeans-init. Weight initialization of deep networks using k-means clustering