shengtianzhou/InferCode

Self_supervised pre-task model (SS-PTM)

Python version I used

Python 3.8.5

3rd party Python libraries used

tqdm 4.59.0

numpy 1.19.2

pytorch 1.4.0

torchvision 0.5.0

torch_scatter 2.0.4

cudatoolkit 10.1

Train a model

sh train.sh

Inference

For usage of the models, please check out the inference.ipynb file, it has an example.