register

nmcardoso · Aug 23, 2023 · b455bcf · b455bcf
1 parent 6abefc5
commit b455bcf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mergernet/jobs/__init__.py b/mergernet/jobs/__init__.py
@@ -1,3 +1,3 @@
 import tensorflow_addons as tfa
 
-tfa.register_all()
+tfa.register_all(custom_kernels=False)