torchdyn:基于PyTorch的万物神经微分方程库

时间:2024-04-08 06:54:44
【文件属性】:

文件名称:torchdyn:基于PyTorch的万物神经微分方程库

文件大小:23.11MB

文件格式:ZIP

更新时间:2024-04-08 06:54:44

control deep-learning neural-network pytorch dynamical-systems

火炬丁 基于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


网友评论