CPU to GPU

docker/compss/runcompss.sh

@@ -6,4 +6,4 @@
 #masterIP=(${Nodes[2]})
 echo "MasterIP is:" $MY_POD_IP
 cd pyeddl
-conda run --no-capture-output -n pyeddl_pycompss_env runcompss -d --lang=python --python_interpreter=python3 --project=/root/project.xml --resources=/root/resources.xml --master_name=$MY_POD_IP eddl_train_batch_compss_mnist_async.py 
\ No newline at end of file
+conda run --no-capture-output -n pyeddl_pycompss_env runcompss --lang=python --python_interpreter=python3 --project=/root/project.xml --resources=/root/resources.xml --master_name=$MY_POD_IP eddl_train_batch_compss_mnist_sync.py 

docker/pyeddl/eddl_compss_distributed.py

@@ -33,7 +33,7 @@ class Eddl_Compss_Distributed:
             eddl.sgd(CVAR_SGD1, CVAR_SGD2),
             losses,
             metrics,
-            eddl.CS_CPU(mem="full_mem"),
+            eddl.CS_GPU(),
             False
         )
 
@@ -79,19 +79,19 @@ class Eddl_Compss_Distributed:
             eddl.sgd(CVAR_SGD1, CVAR_SGD2),
             ["soft_cross_entropy"],
             ["categorical_accuracy"],
-            eddl.CS_CPU(mem="full_mem"),
+            eddl.CS_GPU(),
             False
         )
 
         #print("Modelo built")
-        sys.stdout.flush()
+        #sys.stdout.flush()
 
         # Set the parameters sent from master to the model
         #model.setParameters(net_parametersToTensor(initial_parameters))
         eddl.set_parameters(model, net_parametersToTensor(initial_parameters))
 
         #print("Parametros seteados")
-        sys.stdout.flush()
+        #sys.stdout.flush()
 
         #print(eddl.summary(model))
 
@@ -164,7 +164,7 @@ class Eddl_Compss_Distributed:
             eddl.sgd(CVAR_SGD1, CVAR_SGD2),
             ["soft_cross_entropy"],
             ["categorical_accuracy"],
-            eddl.CS_CPU(mem="full_mem"),
+            eddl.CS_GPU(),
             False
         )
 
@@ -225,7 +225,7 @@ class Eddl_Compss_Distributed:
             eddl.sgd(CVAR_SGD1, CVAR_SGD2),
             ["soft_cross_entropy"],
             ["categorical_accuracy"],
-            eddl.CS_CPU(mem="full_mem"),
+            eddl.CS_GPU(),
             False
         )
 
@@ -274,7 +274,7 @@ class Eddl_Compss_Distributed:
             eddl.sgd(CVAR_SGD1, CVAR_SGD2),
             ["soft_cross_entropy"],
             ["categorical_accuracy"],
-            eddl.CS_CPU(mem="full_mem"),
+            eddl.CS_GPU(),
             False
         )
 

docker/pyeddl/eddl_train_batch_compss_mnist_sync.py

@@ -45,7 +45,7 @@ def main(args):
         eddl.sgd(CVAR_SGD1, CVAR_SGD2),
         ["soft_cross_entropy"],
         ["categorical_accuracy"],
-        eddl.CS_CPU(),
+        eddl.CS_GPU(),
         True
     )
 

docker/pyeddl/examples/eddl_train_batch_compss_cifar10_lenet_async.py

@@ -43,7 +43,7 @@ def main(args):
         eddl.sgd(CVAR_SGD1, CVAR_SGD2),
         ["soft_cross_entropy"],
         ["categorical_accuracy"],
-        eddl.CS_CPU(),
+        eddl.CS_GPU(),
         True if input_file == "" else False
     )
 

docker/pyeddl/examples/eddl_train_batch_compss_cifar10_lenet_sync.py

@@ -46,7 +46,7 @@ def main(args):
         eddl.sgd(CVAR_SGD1, CVAR_SGD2),
         ["soft_cross_entropy"],
         ["categorical_accuracy"],
-        eddl.CS_CPU(),
+        eddl.CS_GPU(),
         True if input_file == "" else False
     )
 

docker/pyeddl/examples/eddl_train_batch_compss_cifar10_lenet_sync_test.py

@@ -42,7 +42,7 @@ def main(args):
         eddl.sgd(CVAR_SGD1, CVAR_SGD2),
         ["soft_cross_entropy"],
         ["categorical_accuracy"],
-        eddl.CS_CPU(),
+        eddl.CS_GPU(),
         True if input_file == "" else False
     )
 

docker/pyeddl/examples/eddl_train_batch_compss_cifar10_vgg16_async.py

@@ -43,7 +43,7 @@ def main(args):
         eddl.sgd(CVAR_SGD1, CVAR_SGD2),
         ["soft_cross_entropy"],
         ["categorical_accuracy"],
-        eddl.CS_CPU(mem="full_mem"),
+        eddl.CS_GPU(mem="full_mem"),
         True
     )
 

docker/pyeddl/examples/eddl_train_batch_compss_cifar10_vgg16_seq.py

@@ -39,7 +39,7 @@ def main(args):
         eddl.sgd(0.001, 0.9),
         ["soft_cross_entropy"],
         ["categorical_accuracy"],
-        eddl.CS_CPU(),
+        eddl.CS_GPU(),
         True
     )
 

docker/pyeddl/examples/eddl_train_batch_compss_cifar10_vgg16_sync.py

@@ -41,7 +41,7 @@ def main(args):
         eddl.sgd(CVAR_SGD1, CVAR_SGD2),
         ["soft_cross_entropy"],
         ["categorical_accuracy"],
-        eddl.CS_CPU(),
+        eddl.CS_GPU(),
         True
     )