未验证 提交 8395d660 编写于 作者: R Roc 提交者: GitHub

Fix paddle setup (#42254)

* expose api

* ref clipgradbynorm

* update

* Update __init__.py
上级 31c33122
...@@ -11,3 +11,8 @@ ...@@ -11,3 +11,8 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
from .gate import GShardGate, BaseGate, SwitchGate, NaiveGate
from .moe_layer import MoELayer
from .grad_clip import ClipGradForMOEByGlobalNorm
ClipGradByGlobalNorm = ClipGradForMOEByGlobalNorm
...@@ -312,6 +312,8 @@ packages=['paddle', ...@@ -312,6 +312,8 @@ packages=['paddle',
'paddle.distributed.auto_parallel.tuner', 'paddle.distributed.auto_parallel.tuner',
'paddle.distributed.auto_parallel.cost', 'paddle.distributed.auto_parallel.cost',
'paddle.distributed.passes', 'paddle.distributed.passes',
'paddle.distributed.models',
'paddle.distributed.models.moe',
'paddle.framework', 'paddle.framework',
'paddle.jit', 'paddle.jit',
'paddle.jit.dy2static', 'paddle.jit.dy2static',
...@@ -366,6 +368,10 @@ packages=['paddle', ...@@ -366,6 +368,10 @@ packages=['paddle',
'paddle.incubate.nn.functional', 'paddle.incubate.nn.functional',
'paddle.incubate.nn.layer', 'paddle.incubate.nn.layer',
'paddle.incubate.optimizer.functional', 'paddle.incubate.optimizer.functional',
'paddle.incubate.distributed',
'paddle.incubate.distributed.models',
'paddle.incubate.distributed.models.moe',
'paddle.incubate.distributed.models.moe.gate',
'paddle.io', 'paddle.io',
'paddle.optimizer', 'paddle.optimizer',
'paddle.nn', 'paddle.nn',
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册