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

support sparse evaluation (#5122)

上级 a794cacc
...@@ -130,9 +130,10 @@ class Trainer(object): ...@@ -130,9 +130,10 @@ 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)
if self.cfg.get('unstructured_prune'): # Unstructured pruner is only enabled in the train mode.
self.pruner = create('UnstructuredPruner')(self.model, if self.cfg.get('unstructured_prune'):
steps_per_epoch) self.pruner = create('UnstructuredPruner')(self.model,
steps_per_epoch)
self._nranks = dist.get_world_size() self._nranks = dist.get_world_size()
self._local_rank = dist.get_rank() self._local_rank = dist.get_rank()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册