Add path to paz module into system before script if script cant find …

…it (PyCharm problem).
Sharda-Tech · Nov 16, 2020 · f16f0c4 · f16f0c4
1 parent 468b348
commit f16f0c4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/examples/discovery_of_latent_keypoints/discover_latent_keypoints.py b/examples/discovery_of_latent_keypoints/discover_latent_keypoints.py
@@ -2,9 +2,13 @@
 # os.environ["CUDA_DEVICE_ORDER"] = 'PCI_BUS_ID'
 # os.environ["CUDA_VISIBLE_DEVICES"] = '0'
 # os.environ["PYOPENGL_PLATFORM"] = 'egl'
+# os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
 import json
 import argparse
 import numpy as np
+# If you cant find the path to paz module (might happen in pycharm), uncomment the following 2 lines:
+# import sys
+# sys.path.insert(1, '/home/robot/git/paz/paz')
 from paz.models import KeypointNetShared
 from paz.models import Projector
 from paz.abstract import GeneratingSequence