Merge pull request #21 from aniketmaurya/master

Updated code to tensorflow 2.x
bckenstler · Jun 5, 2020 · 968cf5a · 968cf5a
2 parents 9d7fb53 + c41751c
commit 968cf5a
Show file tree

Hide file tree

Showing 3 changed files with 624 additions and 141 deletions.
diff --git a/__init__.py b/__init__.py
@@ -0,0 +1 @@
+from .clr_callback import CyclicLR
diff --git a/clr_callback.py b/clr_callback.py
@@ -1,4 +1,6 @@
-from keras.callbacks import *
+from tensorflow.keras.callbacks import *
+from tensorflow.keras import backend as K
+import numpy as np
 
 class CyclicLR(Callback):
     """This callback implements a cyclical learning rate policy (CLR).