n workers test added

Author: stephenoffer

.idea/workspace.xml

@@ -0,0 +1,19 @@
+from tensorflow.keras.applications import ResNet50
+from dkeras import dKeras
+import numpy as np
+import time
+import ray
+
+ray.init()
+data = np.random.uniform(-1, 1, (1000, 224, 224, 3))
+
+for i in range(1, 51):
+    model = dKeras(ResNet50, init_ray=False, wait_for_workers=True, n_workers=i)
+
+    start_time = time.time()
+    preds = model.predict(data)
+    elapsed = time.time() - start_time
+
+    model.close()
+    print("{}\n{} Workers\tTime: {}\tFPS: {}".format('-' * 80, i, elapsed, 1000 / elapsed))
+    time.sleep(3)

testing/json_test.py

@@ -0,0 +1,19 @@
+import time
+import ray
+
+
+@ray.remote(num_cpus=0.1)
+def test():
+    time.sleep(10)
+    return 1
+
+
+def main():
+    ray.init()
+
+    workers = [test.remote() for _ in range(40)]
+    print(ray.get(workers))
+
+
+if __name__ == "__main__":
+    main()

testing/n_worker_test.py

@@ -0,0 +1,12 @@
+model1:
+    file: None
+    CPUs: 1
+    GPUs: 0
+    n_workers: 4
+    ip_address: 192.168.0.102
+model2:
+    file: None
+    CPUs: 4
+    GPUs: 0
+    n_workers: 2
+    ip_address: 192.168.0.102

testing/resource_test.py

@@ -0,0 +1,5 @@
+import yaml
+
+with open('test.yaml', 'r') as f:
+    doc = yaml.load(f, Loader=yaml.FullLoader)
+    print(doc)