Kronecker product decomposition

Kronecker product decomposition in different dataset and model.

Get Start

You can training the Kronecker Model by using main.py

python main.py

if you want to use MNIST Dataset Using LeNet

python main.py -dataset mnist -model LeNet -kron True

if you want to use VIT base 16 Using CIFAR-10

python main.py -dataset cifar10 -model vit_base_16 -kron True

Name		Name	Last commit message	Last commit date
Latest commit History 11 Commits
.vscode		.vscode
data_factorcy		data_factorcy
data_factory		data_factory
gkpd		gkpd
models		models
pics		pics
utils		utils
.gitignore		.gitignore
1		1
1.ipynb		1.ipynb
README.md		README.md
accuracy.pkl		accuracy.pkl
errorbar.png		errorbar.png
kron_out		kron_out
kron_regression.py		kron_regression.py
kron_regression_flops.png		kron_regression_flops.png
linear_regression_flops.png		linear_regression_flops.png
main.py		main.py
metric		metric
one_layer_classification.py		one_layer_classification.py
one_layer_folder.py		one_layer_folder.py
one_layer_group_lasso.py		one_layer_group_lasso.py
one_layer_kron_fold.py		one_layer_kron_fold.py
one_layer_prune.py		one_layer_prune.py
result.md		result.md
run.sh		run.sh
test.py		test.py
train.py		train.py
variance.pkl		variance.pkl