Seems customize_loss_grad is not very self-explanatory and there's is no doc for this argument. I propose to change it to a more clear name like use_default_grad_scale and the default value is true. @reyoung @panyx0718 Need your opinion. Thanks.
|
customize_loss_grad=False): |