1# Copyright 2020 Huawei Technologies Co., Ltd 2# 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14# ============================================================================ 15""" 16network config setting, will be used in test_lenet_quant.py 17""" 18 19from easydict import EasyDict as edict 20 21quant_cfg = edict({ 22 'num_classes': 10, 23 'lr': 0.01, 24 'momentum': 0.9, 25 'epoch_size': 10, 26 'batch_size': 64, 27 'buffer_size': 1000, 28 'image_height': 32, 29 'image_width': 32, 30 'keep_checkpoint_max': 10, 31}) 32