Skip to content
This repository has been archived by the owner on Feb 22, 2020. It is now read-only.

Commit

Permalink
fix(encoder): fix bug in tf inception
Browse files Browse the repository at this point in the history
  • Loading branch information
Larryjianfeng committed Jul 26, 2019
1 parent ded92c5 commit 11dde2b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion gnes/encoder/image/inception.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ def post_init(self):
from .inception_cores.inception_v4 import inception_v4
from .inception_cores.inception_utils import inception_arg_scope
import os
os.environ['CUDA_VISIBLE_DEVICES'] = get_first_available_gpu()
os.environ['CUDA_VISIBLE_DEVICES'] = str(get_first_available_gpu())
g = tf.Graph()
with g.as_default():
arg_scope = inception_arg_scope()
Expand Down

0 comments on commit 11dde2b

Please sign in to comment.