Commit 50e572fa authored by salbiach's avatar salbiach
Browse files

CPU to GPU

parent a72e98cf
......@@ -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
......@@ -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
)
......
......@@ -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
)
......
......@@ -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
)
......
......@@ -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
)
......
......@@ -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
)
......
......@@ -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
)
......
......@@ -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
)
......
......@@ -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
)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment