First, the import didn’t work
from wandb.keras import WandbMetricsLogger, WandbModelCheckpoint
Changed it to:
from wandb.integration.keras import WandbMetricsLogger, WandbModelCheckpoint
Then:
Tracking run with wandb version 0.17.6
TypeError Traceback (most recent call last)
in <cell line: 9>() 51 wandb_callbacks = [ 52 WandbMetricsLogger(), —> 53 WandbModelCheckpoint(filepath=“my_model_{epoch:02d}”), 54 ] 55
/usr/local/lib/python3.10/dist-packages/wandb/integration/keras/callbacks/model_checkpoint.py in init(self, filepath, monitor, verbose, save_best_only, save_weights_only, mode, save_freq, options, initial_value_threshold, **kwargs) 92 **kwargs: Any, 93 ) → None: —> 94 super().init( 95 filepath=filepath, 96 monitor=monitor,
TypeError: ModelCheckpoint.init() got an unexpected keyword argument ‘options’
I’m getting a similar error locally with ubuntu 20.04.