Erva Ulusoy commited on
Commit
126dbe4
·
1 Parent(s): 92ed479

updated requirements

Browse files
Files changed (2) hide show
  1. run_prothgt_app.py +1 -2
  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
- # conv = HGTConv(hidden_channels, hidden_channels, data.metadata(), num_heads, group='sum')
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