Skip to content

Commit

Permalink
Fix comment
Browse files Browse the repository at this point in the history
  • Loading branch information
Anna Mironova committed Jan 26, 2021
1 parent abc23d0 commit 72334ec
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions models/public/netvlad/pre-convert.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,11 +30,11 @@ def main():
parser.add_argument('output_dir', type=Path)
args = parser.parse_args()

tf.disable_eager_execution()

sys.path.append(str(args.input_dir))
nets = importlib.import_module('netvlad_tf.nets')

tf.disable_eager_execution()

tf.reset_default_graph()
image_batch = tf.placeholder(dtype=tf.float32, shape=[None, None, None, 3])
nets.vgg16NetvladPca(image_batch)
Expand Down

0 comments on commit 72334ec

Please sign in to comment.