Skip to content

do not pull in tensorboard's keras callback#336

Open
haohuanw wants to merge 1 commit intogoogle:mainfrom
haohuanw:no-keras-callback
Open

do not pull in tensorboard's keras callback#336
haohuanw wants to merge 1 commit intogoogle:mainfrom
haohuanw:no-keras-callback

Conversation

@haohuanw
Copy link

i am moving some of my models to keras3 and using jax backend for training. i am on a slightly older tensorboard (2.12) so when it pulls in keras callback through from tensorboard.plugins.hparams import api it tries to pull in keras2 since api also import keras callback in tensorboard repo.

this pr basically adjusted import to avoid accidentally pulling in keras in clu.

@haohuanw
Copy link
Author

@andsteing would you be able to help me with reviewing the change? thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant

Comments