1 changed files with 2 additions and 2 deletions
+ 2
- 2
GraphTransformer.py
View File
| @@ -210,9 +210,9 @@ class Hydra(nn.Module): | |||
| return mlp_output | |||
| '''' | |||
| """" | |||
| hala train | |||
| '''' | |||
| """" | |||
| def build_model(gcn_input, model_dim, head): | |||
| model = Hydra(gcn_input, model_dim, head).cuda() | |||
| return model | |||
Loading…