Update retrain script

2019-05-06 03:18:13 +01:00 · 2019-05-06 03:18:13 +01:00 · 3c1b3a4e77
commit 3c1b3a4e77
parent eebeedcdff
1 changed files with 1 additions and 1 deletions
--- a/Logic/TrashRecognition/Retrain.py
+++ b/Logic/TrashRecognition/Retrain.py
@ -1276,7 +1276,7 @@ if __name__ == '__main__':
  parser.add_argument(
      '--bottleneck_dir',
      type=str,
-      default='Model/bottleneck',
+      default='tmp/bottleneck',
      help='Path to cache bottleneck layer values as files.'
  )
  parser.add_argument(