search

DiffEqML / torchdyn

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods
https://torchdyn.org
Apache License 2.0
1.36k stars 125 forks source link

Pretrained SOTA CNF / FFJORD on images #99

Open Zymrael opened 3 years ago

Zymrael commented 3 years ago

We plan to kickstart our pretrained model zoo with weights of a SOTA CNF trained on CIFAR10 and ImageNet.