Browse Source

!4880 GPU modelzoo change Resnet50 default training to Mixprecision

Merge pull request !4880 from VectorSL/model-zoo
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
3cceb64ca5
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      model_zoo/official/cv/resnet/scripts/run_distribute_train_gpu.sh
  2. +1
    -1
      model_zoo/official/cv/resnet/train.py

+ 1
- 1
model_zoo/official/cv/resnet/scripts/run_distribute_train_gpu.sh View File

@@ -55,7 +55,7 @@ then
fi fi




if [ ! -d $PATH2 ]
if [ ! -d $PATH1 ]
then then
echo "error: DATASET_PATH=$PATH1 is not a directory" echo "error: DATASET_PATH=$PATH1 is not a directory"
exit 1 exit 1


+ 1
- 1
model_zoo/official/cv/resnet/train.py View File

@@ -163,7 +163,7 @@ if __name__ == '__main__':
loss = SoftmaxCrossEntropyWithLogits(sparse=True, reduction="mean", is_grad=False, loss = SoftmaxCrossEntropyWithLogits(sparse=True, reduction="mean", is_grad=False,
num_classes=config.class_num) num_classes=config.class_num)


if args_opt.net == "resnet101":
if args_opt.net == "resnet101" or args_opt.net == "resnet50":
opt = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), lr, config.momentum, config.weight_decay, opt = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), lr, config.momentum, config.weight_decay,
config.loss_scale) config.loss_scale)
loss_scale = FixedLossScaleManager(config.loss_scale, drop_overflow_update=False) loss_scale = FixedLossScaleManager(config.loss_scale, drop_overflow_update=False)


Loading…
Cancel
Save