running with python 3.7+ torch 0.4+

yuchaoli / KSE

Exploiting Kernel Sparsity and Entropy for Interpretable CNN Compression

47 stars 21 forks source link

Open Cjkkkk opened 5 years ago

Cjkkkk commented 5 years ago

requires replace async=True with non_blocking=True in file utils/base.py since async is a reserved keyword in python 3.7

Cjkkkk commented 5 years ago

just in case if anyone wants to run code with python3.7+ and torch0.4+ ...