Skip to content

Commit

Permalink
cifar-10: Use unique dir for datasplits (NVIDIA#643)
Browse files Browse the repository at this point in the history
  • Loading branch information
holgerroth authored Jun 6, 2022
1 parent eef008b commit 82fd9ff
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion examples/cifar10/submit_job.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
import argparse
import json
import os
import uuid

from nvflare.fuel.hci.client.fl_admin_api_runner import FLAdminAPIRunner, api_command_wrapper

Expand Down Expand Up @@ -66,10 +67,11 @@ def main():
server_config = read_json(server_config_filename)
meta_config = read_json(meta_config_filename)
print(f"Set alpha to {args.alpha}")
token = str(uuid.uuid4())
job_name = f"{job_name}_alpha{args.alpha}"
server_config["alpha"] = args.alpha
meta_config["name"] = job_name
split_dir = os.path.join(args.train_split_root, job_name)
split_dir = os.path.join(args.train_split_root, f"{job_name}_{token}")
print(f"Set train split root to {split_dir}")
server_config["TRAIN_SPLIT_ROOT"] = split_dir
client_config["TRAIN_SPLIT_ROOT"] = split_dir
Expand Down

0 comments on commit 82fd9ff

Please sign in to comment.