Skip to content

use eqx.combine and jax.vmap in loss_fn to compute detection outputs …#14

Open
bernardev254 wants to merge 1 commit intoZantFoundation:mainfrom
bernardev254:fix/lossfn-use-model-vmap
Open

use eqx.combine and jax.vmap in loss_fn to compute detection outputs …#14
bernardev254 wants to merge 1 commit intoZantFoundation:mainfrom
bernardev254:fix/lossfn-use-model-vmap