From aa18b97cba5e5a8fcb2efb4d6bb98f53b7019517 Mon Sep 17 00:00:00 2001 From: YuanTingHsieh Date: Thu, 5 Sep 2024 14:04:34 -0700 Subject: [PATCH] Fix data save path --- .../data/jobs/hello-pt-cse/app/custom/train.py | 5 ++--- .../data/test_configs/standalone_job/pt_job.yml | 2 +- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/integration_test/data/jobs/hello-pt-cse/app/custom/train.py b/tests/integration_test/data/jobs/hello-pt-cse/app/custom/train.py index 45834de908..c16b1cccbe 100644 --- a/tests/integration_test/data/jobs/hello-pt-cse/app/custom/train.py +++ b/tests/integration_test/data/jobs/hello-pt-cse/app/custom/train.py @@ -27,9 +27,8 @@ from nvflare.app_common.app_constant import ModelName # (optional) set a fixed location so we don't need to download everytime -CIFAR10_ROOT = "/tmp/nvflare/data" - -MODEL_SAVE_PATH_ROOT = "/tmp/nvflare/workdir/cifar10" +CIFAR10_ROOT = "~/data" +MODEL_SAVE_PATH_ROOT = "~/data" # (optional) We change to use GPU to speed things up. # if you want to use CPU, change DEVICE="cpu" diff --git a/tests/integration_test/data/test_configs/standalone_job/pt_job.yml b/tests/integration_test/data/test_configs/standalone_job/pt_job.yml index 7ff00e5949..86964c2c1d 100644 --- a/tests/integration_test/data/test_configs/standalone_job/pt_job.yml +++ b/tests/integration_test/data/test_configs/standalone_job/pt_job.yml @@ -48,6 +48,6 @@ tests: - path: tests.integration_test.src.validators.CrossValResultValidator args: { server_model_names: [ "server" ] } setup: - - python -c "from torchvision.datasets import CIFAR10; CIFAR10(root='/tmp/nvflare/data/', download=True)" + - python -c "from torchvision.datasets import CIFAR10; CIFAR10(root='~/data/', download=True)" teardown: - rm -rf ~/data