search

xiangking / ark-nlp

A private nlp coding package, which quickly implements the SOTA solutions.
Apache License 2.0
310 stars 65 forks source link

fix(utils):修复了pgd中梯度为空时的报错问题 #60

Closed zhw666888 closed 2 years ago

zhw666888 commented 2 years ago

PR types

Fix

PR changes

修复pgd中当梯度为空时候的报错问题。

Description

  1. 在pgd的backup_grad与restore_grad这两个函数中增加判空逻辑，以解决当梯度为空时候的报错问题。