Skip to content

genshengye/Quantization-aware-training

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 

Repository files navigation

Quantization-aware-training

Tensorflow Quantization-aware training

官方介绍

一、需要创建两个graph:

1、一个只用来training,在这个graph上使用create_training_graph().
2、另一个graph,只用来inference,在这个graph上使用create_eval_graph().
3、最后,Freeze这个用来inference的graph,再把它传给toco。
1、train.py,第157行(create_training_graph)
2、freeze.py,第133行(create_eval_graph)

三、注意事项

1、create_training_graph,要出现在cross_entropy之后,tf.train.GradientDescentOptimizer之前。
2、create_eval_graph则是出现在创建完inference graph之后,load checkpoint之前。
3、load checkpoint之后是freeze graph,freeze graph 用的是graph_util.convert_variables_to_constants()

About

Tensorflow Quantization-aware training

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published