NVIDIA / JAX-Toolbox

JAX-Toolbox
Apache License 2.0
210 stars 35 forks source link

Use numpy < 2.0.0 to calculate metrics #915

Closed DwarKapex closed 3 weeks ago

DwarKapex commented 3 weeks ago

TensorFlow has not adopt numpy2 yet, that's why to calculate metrics need to install numpy version less than 2.0.0 to avoid breakage.