From 7836e2e0c4612a777694921c190403da1229b26f Mon Sep 17 00:00:00 2001 From: HengZhang Date: Wed, 27 Nov 2024 20:30:38 +0800 Subject: [PATCH] delete redundant code --- federatedscope/trafficflow/model/FedDGCN.py | 15 --------------- 1 file changed, 15 deletions(-) diff --git a/federatedscope/trafficflow/model/FedDGCN.py b/federatedscope/trafficflow/model/FedDGCN.py index cc78239..697e831 100644 --- a/federatedscope/trafficflow/model/FedDGCN.py +++ b/federatedscope/trafficflow/model/FedDGCN.py @@ -99,18 +99,3 @@ class FedDGCN(nn.Module): output2 = self.end_conv3(output2) return output1 + output2 - - -# Instantiate your model class with config and data -def ModelBuilder(model_config, local_data): - model = FedDGCN(model_config) - return model - - -def call_ddgcrn(model_config, local_data): - if model_config.type == "DDGCRN": - model = ModelBuilder(model_config, local_data) - return model - - -register_model("DDGCRN", call_ddgcrn)