keras/keras_core/callbacks/__init__.py
Gabriel Rasskin 699e4c3174 Tensorboard callback (#211)
* Begin adding testing for tensorboard callback

* Begin protobuf areas

* Sharing WIP PR

* Rework init

* Add start/stop profiling to backend

* Add image/histogram summaries

* Model tracing

* Full tf working

* Enable jax graphs

* Fix cifar100 dataset

* Finish jax integration checkpoint without tracing

* Formatting

* All tests passing locally

* Model path protected, switch to _model

* Remove non-deterministic variable-n tag test

* Formatting

* Remove profiling jax

* Remove tensorboard ops

* Remove profiling

* File utils fix?

* Revert "File utils fix?"

This reverts commit a48ba39a9c51c1e39f4509f795fdc2c316a1da3b.
2023-06-04 13:44:21 -04:00

14 lines
803 B
Python

from keras_core.callbacks.callback import Callback
from keras_core.callbacks.callback_list import CallbackList
from keras_core.callbacks.csv_logger import CSVLogger
from keras_core.callbacks.early_stopping import EarlyStopping
from keras_core.callbacks.history import History
from keras_core.callbacks.lambda_callback import LambdaCallback
from keras_core.callbacks.learning_rate_scheduler import LearningRateScheduler
from keras_core.callbacks.model_checkpoint import ModelCheckpoint
from keras_core.callbacks.progbar_logger import ProgbarLogger
from keras_core.callbacks.reduce_lr_on_plateau import ReduceLROnPlateau
from keras_core.callbacks.remote_monitor import RemoteMonitor
from keras_core.callbacks.tensorboard import TensorBoard
from keras_core.callbacks.terminate_on_nan import TerminateOnNaN