文件名称：mish-cuda
文件大小：28KB
文件格式：ZIP
更新时间：2024-04-07 15:05:33
Cuda
Mish-Cuda：自正则非单调激活函数 这是Diganta Misra（ ）对Mish激活的PyTorch CUDA实现。 安装 目前，它仅作为源分发PyTorch扩展。因此，您需要适当地设置工具链和CUDA编译器进行安装。 工具链-在cxx_linux-64软件包提供了适当的工具链。但是，根据系统，仍然可能存在兼容性问题。您也可以尝试使用系统工具。 CUDA工具包-除驱动程序外，还需要来提供所需的标头和工具。从nVidia获取适合您Linux发行版的版本，否则请查看发行版的特定说明。 重要的是您的CUDA Toolkit必须与PyTorch所构建的版本相匹配，否则可能会发生错误。目前，PyTorch为v10.0和v9.2构建。 表现 CUDA实施似乎反映了原始实施的学习表现，并且未观察到稳定性问题。就功能的速度而言，它可以与其他PyTorch激活功能相当，并且比纯PyTorch实施
【文件预览】：
mish-cuda-master
----.dockerignore(51B)
----src()
--------mish_cuda()
----Dockerfile(794B)
----LICENSE(1KB)
----test()
--------test_mish.py(3KB)
--------perftest.py(4KB)
----setup.py(898B)
----README.md(3KB)
----csrc()
--------cpu()
--------cuda()
----.gitignore(1KB)
----extra()
--------Performance.ipynb(24KB)
--------MiniTest.ipynb(6KB)
--------error_callback.py(2KB)
--------Derivatives.ipynb(12KB)