Spaces:
Sleeping
Sleeping
Erva Ulusoy
commited on
Commit
·
126dbe4
1
Parent(s):
92ed479
updated requirements
Browse files- run_prothgt_app.py +1 -2
- setup.sh +1 -1
run_prothgt_app.py
CHANGED
@@ -18,8 +18,7 @@ class ProtHGT(torch.nn.Module):
|
|
18 |
|
19 |
self.convs = torch.nn.ModuleList()
|
20 |
for _ in range(num_layers):
|
21 |
-
|
22 |
-
conv = HGTConv(hidden_channels, hidden_channels, data.metadata(), num_heads)
|
23 |
self.convs.append(conv)
|
24 |
|
25 |
self.mlp = MLP(mlp_hidden_layers , dropout=mlp_dropout, norm=None)
|
|
|
18 |
|
19 |
self.convs = torch.nn.ModuleList()
|
20 |
for _ in range(num_layers):
|
21 |
+
conv = HGTConv(hidden_channels, hidden_channels, data.metadata(), num_heads, group='sum')
|
|
|
22 |
self.convs.append(conv)
|
23 |
|
24 |
self.mlp = MLP(mlp_hidden_layers , dropout=mlp_dropout, norm=None)
|
setup.sh
CHANGED
@@ -3,4 +3,4 @@ pip uninstall torch-scatter torch_sparse torch_geometric torch_cluster torch_geo
|
|
3 |
# pip install torch-scatter -f https://data.pyg.org/whl/torch-1.12.0+cpu.html
|
4 |
pip install torch_sparse -f https://data.pyg.org/whl/torch-1.12.0+cpu.html
|
5 |
# pip install torch-cluster -f https://data.pyg.org/whl/torch-1.12.0+cpu.html
|
6 |
-
pip install torch_geometric
|
|
|
3 |
# pip install torch-scatter -f https://data.pyg.org/whl/torch-1.12.0+cpu.html
|
4 |
pip install torch_sparse -f https://data.pyg.org/whl/torch-1.12.0+cpu.html
|
5 |
# pip install torch-cluster -f https://data.pyg.org/whl/torch-1.12.0+cpu.html
|
6 |
+
pip install torch_geometric==2.2.0
|