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)