search

applied-ml-bde / bde

Bayesian Deep Ensembles
https://applied-ml-bde.github.io/bde/
BSD 3-Clause "New" or "Revised" License
2 stars 0 forks source link

Training fix #16

Closed cklyne closed 4 months ago

cklyne commented 4 months ago

This should solve the issue with the training-method implementation (checked via testing).

cklyne commented 4 months ago

I'm closing without a review since the merge is not into main. The review will be done when basic_jax is merged into main.