您好,欢迎光临本网站![请登录][注册会员]  

搜索资源列表

  1. fastmoe:PyTorch的快速MoE提示-源码

  2. || 介绍 一个易于使用且高效的系统,可支持PyTorch的专家混合(MoE)模型。 安装 先决条件 带有CUDA的PyTorch是必需的。 该存储库目前已通过PyTorch v1.8.0和CUDA 10进行了测试,并具有对较早版本的设计兼容性。 如果启用了分布式专家功能,则需要具有P2P通信支持的NCCL,通常版本>=2.7.5 。 正在安装 FastMoE包含一组PyTorch定制的运算符,包括C和Python组件。 使用python setup.py install可以轻松安装并享
  3. 所属分类:其它

    • 发布日期:2021-03-07
    • 文件大小:396kb
    • 提供者:weixin_42098104