realize LeNet5 in pytorch, net.py is origin network, net_quant.py is a network quantified on the basis of net, you should first use train.py to train a model of net, then use linear_quant function in net_quant.py to quantify the model, finally you can use test_quant.py to validate the quantized model.
-
Notifications
You must be signed in to change notification settings - Fork 1
bird1and1fish/LeNet5
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
realize LeNet5 in pytorch
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published