patglow

环境：pytorch：1.10.1，torchvision:0.11.2

从main.py文件的主函数进入，具体的训练过程main.py里面有注释说明

glow_utils文件夹里是glow模型需要调用的python文件

utils文件夹是patdnn模型需要调用的python文件

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
__pycache__		__pycache__
glow_utils		glow_utils
models		models
util		util
utils		utils
README.md		README.md
check.txt		check.txt
dataset.py		dataset.py
main.py		main.py
main_3.py		main_3.py
main_check.py		main_check.py
main_loss3.py		main_loss3.py
main_mnist.py		main_mnist.py
main_no_admm.py		main_no_admm.py
main_resnet_patdnn.py		main_resnet_patdnn.py
main_resnet_swp.py		main_resnet_swp.py
main_retrain.py		main_retrain.py
main_swp.py		main_swp.py
pattern_set.npy		pattern_set.npy
pattern_set_glow.npy		pattern_set_glow.npy
pattern_set_imagenet.npy		pattern_set_imagenet.npy
pattern_set_mnist.npy		pattern_set_mnist.npy
prune_main.py		prune_main.py
re_train.py		re_train.py
reader.py		reader.py
run_script.py		run_script.py

Provide feedback