delete redundant code
This commit is contained in:
parent
482e8d562d
commit
7836e2e0c4
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Reference in New Issue