diff --git a/keras_contrib/callbacks/__init__.py b/keras_contrib/callbacks/__init__.py new file mode 100644 index 0000000..a64ec7e --- /dev/null +++ b/keras_contrib/callbacks/__init__.py @@ -0,0 +1 @@ +from .snapshot import SnapshotCallbackBuilder, SnapshotModelCheckpoint diff --git a/keras_contrib/callbacks.py b/keras_contrib/callbacks/snapshot.py similarity index 100% rename from keras_contrib/callbacks.py rename to keras_contrib/callbacks/snapshot.py diff --git a/tests/keras_contrib/callbacks_test.py b/tests/keras_contrib/callbacks/snapshot_test.py similarity index 100% rename from tests/keras_contrib/callbacks_test.py rename to tests/keras_contrib/callbacks/snapshot_test.py