From 2b52a3b54aaff7564ecce78f4ca660c240d919a5 Mon Sep 17 00:00:00 2001 From: matthieudelaro Date: Wed, 29 Jun 2016 22:09:04 +0900 Subject: [PATCH 1/2] Added nut.yml file. Apply transformation with 'nut run'. See github.com/matthieudelaro/nut --- nut.yml | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 nut.yml diff --git a/nut.yml b/nut.yml new file mode 100644 index 0000000..5864bf8 --- /dev/null +++ b/nut.yml @@ -0,0 +1,19 @@ +syntax_version: "7" +project_name: neural-style +docker_image: matthieudelaro/cuda-torch-plus +container_working_directory: /opt/style +volumes: + main: + host_path: . + container_path: /opt/style +macros: + setup: + usage: download model + actions: + - sh ./models/download_models.sh + run: + usage: run the neural network in the container, on GPU + enable_nvidia_devices: true + actions: + - export iterations=1000 + - time th neural_style.lua -num_iterations $iterations -seed 123 -style_image style_image.jpg -content_image content_image.jpg -output_image "output_image_`date '+%Y-%m-%d_%Hh%Mmin%S'`.png" From 623175182963d25097b853475e03bac052bfe537 Mon Sep 17 00:00:00 2001 From: matthieudelaro Date: Sat, 2 Jul 2016 18:35:39 +0900 Subject: [PATCH 2/2] Update nut.yml --- nut.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nut.yml b/nut.yml index 5864bf8..ed89a1e 100644 --- a/nut.yml +++ b/nut.yml @@ -2,6 +2,7 @@ syntax_version: "7" project_name: neural-style docker_image: matthieudelaro/cuda-torch-plus container_working_directory: /opt/style +enable_nvidia_devices: true volumes: main: host_path: . @@ -13,7 +14,6 @@ macros: - sh ./models/download_models.sh run: usage: run the neural network in the container, on GPU - enable_nvidia_devices: true actions: - export iterations=1000 - time th neural_style.lua -num_iterations $iterations -seed 123 -style_image style_image.jpg -content_image content_image.jpg -output_image "output_image_`date '+%Y-%m-%d_%Hh%Mmin%S'`.png"