예제 #1
0
 def _create_global_step(self):
     if self.weight_quantize_type == 'range_abs_max' or \
         self.activation_quantize_type == 'range_abs_max':
         self.global_step = autoincreased_step_counter()
예제 #2
0
def _decay_step_counter(begin=0):
    # the first global step is zero in learning rate decay
    global_step = nn.autoincreased_step_counter(
        counter_name='@LR_DECAY_COUNTER@', begin=begin, step=1)
    global_step = tensor.cast(global_step, 'float32')
    return global_step