|
|
@@ -1,4 +1,6 @@
|
|
|
import argparse
|
|
|
+from applications.fedssl.client_with_pgfed import FedSSLWithPgFedClient
|
|
|
+from applications.fedssl.server_with_pgfed import FedSSLWithPgFedServer
|
|
|
|
|
|
import easyfl
|
|
|
from client import FedSSLClient
|
|
|
@@ -162,8 +164,8 @@ def run():
|
|
|
|
|
|
model = get_model(args.model, args.encoder_network, args.predictor_network)
|
|
|
easyfl.register_model(model)
|
|
|
- easyfl.register_client(FedSSLClient)
|
|
|
- easyfl.register_server(FedSSLServer)
|
|
|
+ easyfl.register_client(FedSSLWithPgFedClient)
|
|
|
+ easyfl.register_server(FedSSLWithPgFedServer)
|
|
|
easyfl.init(config, init_all=True)
|
|
|
easyfl.run()
|
|
|
|
