-
torchdyn：一个适用于所有事物的基于PyTorch的神经微分方程库
资源介绍
火炬丁
基于PyTorch的万物神经微分方程库。由维护。
持续集成
系统/ Python版本
3.6
3.7
3.8+
Ubuntu 16.04
Ubuntu 18.04
视窗
快速开始
神经微分方程变得简单：
from torchdyn import NeuralODE
# your preferred torch.nn.Module here
f = nn.Sequential(nn.Conv2d(1, 32, 3),
nn.Softplus(),
nn.Conv2d(32, 1, 3)
)
nde = NeuralODE(f)
而且你很好。 nde对象可以与其他深度学习模型无缝组合。
安装
稳定发布：
pip install torchdyn
注意：暂时需要额外的手动安装tor