Remove int32 cast from StructureExporter.

PiperOrigin-RevId: 252976884
google-research · Jun 13, 2019 · 52f4fac · 52f4fac
1 parent 85f3ca7
commit 52f4fac
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/morph_net/tools/structure_exporter.py b/morph_net/tools/structure_exporter.py
@@ -70,8 +70,7 @@ def __init__(self,
         tf.logging.warning('No regularizer found for: %s', op.name)
         continue
 
-      self._tensors[rename_fn(op.name)] = tf.cast(op_regularizer.alive_vector,
-                                                  tf.int32)
+      self._tensors[rename_fn(op.name)] = op_regularizer.alive_vector
 
   @property
   def tensors(self):