czzhangheng
/
FS-TFP
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

delete redundant code

This commit is contained in:
HengZhang 2024-11-27 20:30:38 +08:00
parent 482e8d562d
commit 7836e2e0c4
1 changed files with 0 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
federatedscope/trafficflow/model/FedDGCN.py
View File

@ -99,18 +99,3 @@ class FedDGCN(nn.Module):
output2 = self.end_conv3(output2) output2 = self.end_conv3(output2)
return output1 + 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)