提交 75eb04c5 编写于 作者: M Megvii Engine Team 提交者: huangxinda

feat(mge/experimental): add WeightScaler support

GitOrigin-RevId: a1d5b68498e51bf1f7dc2266b72c19d17c130195
上级 dedecf69
# -*- coding: utf-8 -*-
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
......@@ -1462,6 +1462,5 @@ warp_affine = deprecated_func("1.3", "megengine.functional.vision", "warp_affine
warp_perspective = deprecated_func(
"1.3", "megengine.functional.vision", "warp_perspective", True
)
from .loss import * # isort:skip
from .quantized import conv_bias_activation # isort:skip
from .loss import * # isort:skip
#!/usr/bin/env bash
set -e
cd $(dirname $0)/..
ISORT_ARG=""
BLACK_ARG=""
while getopts 'd' OPT; do
case $OPT in
d)
ISORT_ARG="--diff --check-only"
BLACK_ARG="--diff --check"
;;
?)
echo "Usage: `basename $0` [-d]"
esac
done
# do not isort megengine/__init__.py file, caused we must
# init library load path before load dependent lib in core
isort $ISORT_ARG -j $(nproc) -rc megengine test examples -s megengine/__init__.py
black $BLACK_ARG --target-version=py35 -- megengine test examples
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册