FluxML / FluxBench.jl

Benchmarks for the FluxML ecosystem for deep learning, scientific machine learning, differentiable programming etc including AD and CUDA accelerated workloads
https://speed.fluxml.ai
14 stars 9 forks source link

Cleanup some bechmarking #4

Closed DhairyaLGandhi closed 3 years ago

DhairyaLGandhi commented 3 years ago

Disables most tests temporarily so that its easier to see results

DhairyaLGandhi commented 3 years ago

Okay this seems to be working. Thanks @maleadt for all the help!

DhairyaLGandhi commented 3 years ago

This PR has gotten pretty big, so the DiffEqFlux tests can be done in the next one. It seems to fail with FFJORD not being GPU compatible. I get the Cannot take CPU address... error.