fix norm_by_batchsize

pull/822/head
Junkun 3 years ago
parent 91bc5959a9
commit 064d42137c

@ -46,7 +46,7 @@ class CTCLoss(nn.Layer):
if grad_norm_type == 'instance': if grad_norm_type == 'instance':
self.norm_by_times = True self.norm_by_times = True
if grad_norm_type == 'batch': if grad_norm_type == 'batch':
self.norm_by_times = True self.norm_by_batchsize = True
if grad_norm_type == 'frame': if grad_norm_type == 'frame':
self.norm_by_total_logits_len = True self.norm_by_total_logits_len = True

Loading…
Cancel
Save