search

mlr-org / mlr3keras

Deep learning for mlr3
GNU Lesser General Public License v3.0
36 stars 2 forks source link

Flexibility wrt tf and keras versions #41

Open pfistfl opened 3 years ago

pfistfl commented 3 years ago

Support multiple versions of keras and tensorflow

hrvg commented 3 years ago

Thanks for this package; it looks extremely promising.

I was wondering if you had any update concerning the support of other tensorflow and keras versions. I am unfamiliar with what could be a major roadblock, but I hit a significant hurdle in my workflow as my system cannot seamlessly install CUDA 10.1 which is required by tensorflow-2.3.0. I have working installs of keras and tensorflow which default to 2.5 and CUDA 11.4.

This would be a significant improvement considering the current defaults from keras and tensorflow and how tedious it is to resort to docker containers or building from source.