From 52a238f4bb810d022f5f08ed2550aef00f4f81bd Mon Sep 17 00:00:00 2001 From: caojian05 Date: Tue, 28 Apr 2020 19:38:43 +0800 Subject: [PATCH] remove enable mixed precision for ge --- example/vgg16_cifar10/run_distribute_train.sh | 1 + example/vgg16_cifar10/train.py | 3 +-- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/example/vgg16_cifar10/run_distribute_train.sh b/example/vgg16_cifar10/run_distribute_train.sh index c9b8dfc48f..9cd8c63db6 100755 --- a/example/vgg16_cifar10/run_distribute_train.sh +++ b/example/vgg16_cifar10/run_distribute_train.sh @@ -36,6 +36,7 @@ ulimit -u unlimited export DEVICE_NUM=8 export RANK_SIZE=8 export MINDSPORE_HCCL_CONFIG_PATH=$1 +export RANK_TABLE_FILE=$1 for((i=0; i<${DEVICE_NUM}; i++)) do diff --git a/example/vgg16_cifar10/train.py b/example/vgg16_cifar10/train.py index 234e3f7c7e..b71bb502b6 100644 --- a/example/vgg16_cifar10/train.py +++ b/example/vgg16_cifar10/train.py @@ -83,8 +83,7 @@ if __name__ == '__main__': lr = lr_steps(0, lr_max=cfg.lr_init, total_epochs=cfg.epoch_size, steps_per_epoch=batch_num) opt = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), Tensor(lr), cfg.momentum, weight_decay=cfg.weight_decay) loss = nn.SoftmaxCrossEntropyWithLogits(sparse=True, reduction='mean', is_grad=False) - model = Model(net, loss_fn=loss, optimizer=opt, metrics={'acc'}, - amp_level="O2", keep_batchnorm_fp32=False, loss_scale_manager=None) + model = Model(net, loss_fn=loss, optimizer=opt, metrics={'acc'}) config_ck = CheckpointConfig(save_checkpoint_steps=batch_num * 5, keep_checkpoint_max=cfg.keep_checkpoint_max) time_cb = TimeMonitor(data_size=batch_num)