From 9e2ec3b8d8d5b63f798465b65d7764882ad9b722 Mon Sep 17 00:00:00 2001 From: leilei_snow Date: Wed, 22 Apr 2020 18:26:08 +0800 Subject: [PATCH] check the legal value of weight_decay and loss_scale --- mindspore/nn/optim/optimizer.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/mindspore/nn/optim/optimizer.py b/mindspore/nn/optim/optimizer.py index 719e7aa55e..72593e8001 100755 --- a/mindspore/nn/optim/optimizer.py +++ b/mindspore/nn/optim/optimizer.py @@ -88,14 +88,12 @@ class Optimizer(Cell): if isinstance(weight_decay, int): weight_decay = float(weight_decay) - if not isinstance(weight_decay, float): - raise TypeError("weight_decay should be a float number!") + validator.check_float_legal_value('weight_decay', weight_decay, None) if isinstance(loss_scale, int): loss_scale = float(loss_scale) - if not isinstance(loss_scale, float): - raise TypeError("loss_scale should be a float number!") + validator.check_float_legal_value('loss_scale', loss_scale, None) if loss_scale <= 0.0: raise ValueError("Loss scale should be greater than 0, but got {}".format(loss_scale))