fixes

SiLiKhon · Oct 14, 2021 · 9b6313c · 9b6313c
1 parent 72e9c5e
commit 9b6313c
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 2 deletions.
diff --git a/.gitattributes b/.gitattributes
@@ -0,0 +1,3 @@
+* text=auto
+
+*.sh text eol=crlf
diff --git a/model_export/dump_graph.py b/model_export/dump_graph.py
@@ -7,7 +7,8 @@
 from tensorflow.core.framework import attr_value_pb2
 from tensorflow.python.tools import optimize_for_inference_lib
 
-import tf2xla_pb2
+
+from . import tf2xla_pb2
 
 
 def model_to_graph(

diff --git a/model_export/run_docker_centos.sh b/model_export/run_docker_centos.sh
@@ -2,5 +2,6 @@ docker run -it \
   --rm \
   --env HOME=`pwd` \
   -v `pwd`:`pwd` \
+  --network host \
   centos:centos7.8.2003 \
   /bin/bash
diff --git a/models/nn.py b/models/nn.py
@@ -1,5 +1,6 @@
 import tensorflow as tf
-
+# DO NOT REMOVE. used by code inside eval()
+import numpy as np  # noqa: F401
 
 custom_objects = {}