Add evaluation script, code review and fix some bugs

1. Add new `train_all` method for one shot calling
2. Print time used in 1k iterations
3. Correct label dimension in predict function
4. Transpose distance matrix for convenient indexing
5. Sort dictionary before generate signature
6. Extract visible CUDA setting function

1 files changed, 10 insertions, 0 deletions

diff --git a/utils/misc.py b/utils/misc.py
new file mode 100644
index 0000000..b850830
--- /dev/null
+++ b/utils/misc.py
@@ -0,0 +1,10 @@
+import os
+
+from utils.configuration import SystemConfiguration
+
+
+def set_visible_cuda(config: SystemConfiguration):
+    """Set environment variable CUDA device(s)"""
+    CUDA_VISIBLE_DEVICES = config.get('CUDA_VISIBLE_DEVICES', None)
+    if CUDA_VISIBLE_DEVICES:
+        os.environ['CUDA_VISIBLE_DEVICES'] = CUDA_VISIBLE_DEVICES