Support GPU training in the NVFlare demo (#7965)

2022-06-02 06:52:36 -07:00
parent 6b55150e80
commit 31e6902e43
4 changed files with 26 additions and 3 deletions
--- a/demo/nvflare/config/config_fed_client.json
+++ b/demo/nvflare/config/config_fed_client.json
@@ -12,7 +12,8 @@
          "world_size": 2,
          "server_cert_path": "server-cert.pem",
          "client_key_path": "client-key.pem",
-          "client_cert_path": "client-cert.pem"
+          "client_cert_path": "client-cert.pem",
+          "use_gpus": "false"
        }
      }
    }