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