未验证 提交 3233ef34 编写于 作者: M minghaoBD 提交者: GitHub

support sparse evaluation (#5122)

上级 a794cacc
...@@ -130,6 +130,7 @@ class Trainer(object): ...@@ -130,6 +130,7 @@ class Trainer(object):
self.lr = create('LearningRate')(steps_per_epoch) self.lr = create('LearningRate')(steps_per_epoch)
self.optimizer = create('OptimizerBuilder')(self.lr, self.model) self.optimizer = create('OptimizerBuilder')(self.lr, self.model)
# Unstructured pruner is only enabled in the train mode.
if self.cfg.get('unstructured_prune'): if self.cfg.get('unstructured_prune'):
self.pruner = create('UnstructuredPruner')(self.model, self.pruner = create('UnstructuredPruner')(self.model,
steps_per_epoch) steps_per_epoch)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册