Simon
/
graph-barlow-twins
mirror of https://github.com/pbielak/graph-barlow-twins


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
13

	
14

	
15

	
16

	
17

	
18

	
19

	
20

	
21

	
22

	
23

	
24

	
25

	
26

	
27

	
28

	
29

	
            # General packages
numpy==1.20.1
pandas==1.2.2
scikit_learn==0.24.1
seaborn==0.11.2
tqdm==4.57.0
PyYAML==5.4.1
jupyter==1.0.0
matplotlib==3.3.1

# DVC
dvc[s3]==2.1.0

# PyTorch Geometric
--find-links https://download.pytorch.org/whl/torch_stable.html
--find-links https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html

torch==1.7.0+cu110
torch-scatter==2.0.5
torch-sparse==0.6.8
torch-cluster==1.5.8
torch-spline-conv==1.2.0
torch-geometric==1.7.0

# Open Graph Benchmark
ogb>=1.3.0

# Lightning-bolts for LR scheduler
lightning-bolts==0.3.3