|
- # Copyright 2021 Huawei Technologies Co., Ltd
- #
- # Licensed under the Apache License, Version 2.0 (the "License");
- # you may not use this file except in compliance with the License.
- # You may obtain a copy of the License at
- #
- # http://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- # ============================================================================
- """
- network config setting, will be used in main.py
- """
- from easydict import EasyDict as edict
-
- config_gpu = edict({
- 'is_save_on_master': False,
-
- 'batch_size': 128,
- 'epoch_size': 100,
- 'num_classes': 1000,
- 'work_nums': 8,
-
- 'loss_scale': 1024,
- 'smooth_factor': 0.1,
- 'weight_decay': 0.00001,
- 'optim': "momentum",
- 'momentum': 0.9,
- 'amp_level': 'O2',
- 'decay': 0.9,
- 'epsilon': 1.0,
-
- 'keep_checkpoint_max': 10,
- 'save_checkpoint_epochs': 1,
-
- 'lr_init': 0.0001,
- 'lr_end': 0.000004,
- 'lr_max': 1.,
- 'warmup_epochs': 2,
- 'start_epoch': 1,
-
- # hardware
- 'device_target': "GPU"
- })
-
- config_ascend = edict({
- 'is_save_on_master': False,
-
- 'batch_size': 128,
- 'epoch_size': 250,
- 'num_classes': 1000,
- 'work_nums': 8,
-
- 'loss_scale': 1024,
- 'smooth_factor': 0.1,
- 'weight_decay': 0.00004,
- 'momentum': 0.9,
- 'optim': "rmsprop",
- 'amp_level': 'O3',
- 'decay': 0.9,
- 'epsilon': 1.0,
-
- 'keep_checkpoint_max': 10,
- 'save_checkpoint_epochs': 10,
-
- 'lr_init': 0.00004,
- 'lr_end': 0.000004,
- 'lr_max': 0.4,
- 'warmup_epochs': 1,
- 'start_epoch': 1,
- # hardware
- 'device_target': "Ascend"
- })
|