Я пытаюсь переобучить модель мобильной сети, используя -
!python tensorflow/hubmaster/examples/image_retraining/retrain.py \
--tfhub_module=https://tfhub.dev/google/imagenet/mobilenet_v2_100_224/feature_vector/2
Но получаю следующую ошибку -
Traceback (most recent call last):
File "tensorflow/tensorflow/hub-master/examples/image_retraining/retrain.py", line 1313, in <module>
tf.app.run(main=main, argv=[sys.argv[0]] + unparsed)
File "/anaconda3/lib/python3.6/site-packages/tensorflow/python/platform/app.py", line 48, in run
_sys.exit(main(_sys.argv[:1] + flags_passthrough))
File "tensorflow/tensorflow/hub-master/examples/image_retraining/retrain.py", line 998, in main
module_spec = hub.load_module_spec(FLAGS.tfhub_module)
File "/anaconda3/lib/python3.6/site-packages/tensorflow_hub/module.py", line 57, in load_module_spec
path = registry.resolver(path)
File "/anaconda3/lib/python3.6/site-packages/tensorflow_hub/registry.py", line 42, in __call__
return impl(*args, **kwargs)
File "/anaconda3/lib/python3.6/site-packages/tensorflow_hub/compressed_module_resolver.py", line 77, in __call__
module_dir = _module_dir(handle)
File "/anaconda3/lib/python3.6/site-packages/tensorflow_hub/compressed_module_resolver.py", line 45, in _module_dir
cache_dir = resolver.tfhub_cache_dir(use_temp=True)
File "/anaconda3/lib/python3.6/site-packages/tensorflow_hub/resolver.py", line 68, in tfhub_cache_dir
os.getenv(_TFHUB_CACHE_DIR, "") or FLAGS["tfhub_cache_dir"].value or
TypeError: '_FlagValues' object is not subscriptable
Я пытался изменить os.getenv(_TFHUB_CACHE_DIR, "")
на os.getenv(TFHUB_CACHE_DIR, "")
в resolver.py
(вторая последняя строка ошибки), как показано здесь , но я все еще получаю ту же ошибку