__init__.py 2.8 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
#   Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
#
# 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.

# TODO: import framework api under this directory 
H
hong 已提交
16
__all__ = [
17
    'create_global_var', 'create_parameter', 'ParamAttr', 'Variable',
H
hong 已提交
18 19
    'CPUPlace', 'CUDAPlace', 'CUDAPinnedPlace'
]
20

21 22 23 24 25 26 27 28 29 30
__all__ += [
    'BackwardStrategy', 'grad', 'LayerList', 'load', 'save', 'prepare_context',
    'to_variable', 'no_grad', 'ParallelEnv', 'DataParallel'
]

__all__ += [
    'NoamDecay', 'PiecewiseDecay', 'NaturalExpDecay', 'ExponentialDecay',
    'InverseTimeDecay', 'PolynomialDecay', 'CosineDecay'
]

31 32
from . import random
from .random import manual_seed
33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62

from ..fluid.framework import Variable  #DEFINE_ALIAS
from ..fluid.param_attr import ParamAttr  #DEFINE_ALIAS
from ..fluid.layers.tensor import create_global_var  #DEFINE_ALIAS
from ..fluid.layers.tensor import create_parameter  #DEFINE_ALIAS
from ..fluid.core import CPUPlace  #DEFINE_ALIAS
from ..fluid.core import CUDAPlace  #DEFINE_ALIAS
from ..fluid.core import CUDAPinnedPlace  #DEFINE_ALIAS

from paddle.fluid import core  #DEFINE_ALIAS
from ..fluid.dygraph.base import no_grad  #DEFINE_ALIAS
from ..fluid.dygraph.base import to_variable  #DEFINE_ALIAS
from ..fluid.dygraph.base import grad  #DEFINE_ALIAS
from ..fluid.dygraph.checkpoint import load_dygraph  #DEFINE_ALIAS
from ..fluid.dygraph.checkpoint import save_dygraph  #DEFINE_ALIAS
from ..fluid.dygraph.checkpoint import load_dygraph as load  #DEFINE_ALIAS
from ..fluid.dygraph.checkpoint import save_dygraph as save  #DEFINE_ALIAS
from ..fluid.dygraph.parallel import prepare_context  #DEFINE_ALIAS
from ..fluid.dygraph.parallel import ParallelEnv  #DEFINE_ALIAS
from ..fluid.dygraph.parallel import DataParallel  #DEFINE_ALIAS

from ..fluid.dygraph.learning_rate_scheduler import NoamDecay  #DEFINE_ALIAS
from ..fluid.dygraph.learning_rate_scheduler import PiecewiseDecay  #DEFINE_ALIAS
from ..fluid.dygraph.learning_rate_scheduler import NaturalExpDecay  #DEFINE_ALIAS
from ..fluid.dygraph.learning_rate_scheduler import ExponentialDecay  #DEFINE_ALIAS
from ..fluid.dygraph.learning_rate_scheduler import InverseTimeDecay  #DEFINE_ALIAS
from ..fluid.dygraph.learning_rate_scheduler import PolynomialDecay  #DEFINE_ALIAS
from ..fluid.dygraph.learning_rate_scheduler import CosineDecay  #DEFINE_ALIAS

BackwardStrategy = core.BackwardStrategy