karl1018 / eml_pruning_admm Goto Github PK
View Code? Open in Web Editor NEWThis project forked from fffrozen090/eml_pruning_admm
Final Project for Embedded Machine Learning of Heidelberg University. Task of Group5 is replicating the ADMM pruning and implement it specifically with LeNet-5 and AlexNet. This fork is the LeNet part.
License: MIT License