diff --git a/examples/monai-2D-mednist/README.rst b/examples/monai-2D-mednist/README.rst index f88f3d31d..6b3d91e26 100644 --- a/examples/monai-2D-mednist/README.rst +++ b/examples/monai-2D-mednist/README.rst @@ -91,7 +91,7 @@ Connecting clients: export FEDN_PACKAGE_EXTRACT_DIR=package export FEDN_DATA_PATH=/data/ export FEDN_CLIENT_SETTINGS_PATH=/client_settings.yaml - export export FEDN_DATA_SPLIT_INDEX=0 + export FEDN_DATA_SPLIT_INDEX=0 fedn client start -in client.yaml --secure=True --force-ssl @@ -105,9 +105,11 @@ For convenience, there is a Docker image hosted on ghrc.io with fedn preinstalle docker run \ -v $PWD/client.yaml:/app/client.yaml \ -v $PWD/client_settings.yaml:/app/client_settings.yaml \ + -v $PWD/data:/app/data \ -e FEDN_PACKAGE_EXTRACT_DIR=package \ - -e FEDN_DATA_PATH=./data/ \ + -e FEDN_DATA_PATH=/app/data/ \ -e FEDN_CLIENT_SETTINGS_PATH=/app/client_settings.yaml \ + -e FEDN_DATA_SPLIT_INDEX=0 \ ghcr.io/scaleoutsystems/fedn/fedn:0.9.0 run client -in client.yaml --force-ssl --secure=True diff --git a/examples/monai-2D-mednist/client/train.py b/examples/monai-2D-mednist/client/train.py index 8fc4b05b7..f590fd4f7 100644 --- a/examples/monai-2D-mednist/client/train.py +++ b/examples/monai-2D-mednist/client/train.py @@ -73,7 +73,7 @@ def train(in_model_path, out_model_path, data_path=None, client_settings_path=No image_list = clients["client " + str(split_index)]["train"] - train_ds = MedNISTDataset(data_path=data_path, transforms=train_transforms, image_files=image_list) + train_ds = MedNISTDataset(data_path=data_path+'/MedNIST/', transforms=train_transforms, image_files=image_list) train_loader = DataLoader(train_ds, batch_size=batch_size, shuffle=True, num_workers=num_workers) # Load parmeters and initialize model diff --git a/examples/monai-2D-mednist/client/validate.py b/examples/monai-2D-mednist/client/validate.py index a3053c119..61684867c 100644 --- a/examples/monai-2D-mednist/client/validate.py +++ b/examples/monai-2D-mednist/client/validate.py @@ -55,7 +55,7 @@ def validate(in_model_path, out_json_path, data_path=None, client_settings_path= image_list = clients["client " + str(split_index)]["validation"] - val_ds = MedNISTDataset(data_path="data/MedNIST", transforms=val_transforms, image_files=image_list) + val_ds = MedNISTDataset(data_path=data_path+"/MedNIST/", transforms=val_transforms, image_files=image_list) val_loader = DataLoader(val_ds, batch_size=batch_size, shuffle=True, num_workers=num_workers)