About semisupervised_MNIST_CIFAR10\pre-training\main_SBMs_node_classification.py

Shen-Lab / GraphCL

[NeurIPS 2020] "Graph Contrastive Learning with Augmentations" by Yuning You, Tianlong Chen, Yongduo Sui, Ting Chen, Zhangyang Wang, Yang Shen

MIT License

559 stars 103 forks source link

About semisupervised_MNIST_CIFAR10\pre-training\main_SBMs_node_classification.py #31

Open liuxy1103 opened 3 years ago

liuxy1103 commented 3 years ago

There are no the script "from train.train_SBMs_node_classification import train_epoch, evaluate_network"

liuxy1103 commented 3 years ago

Would you mind providing the script for us?

yyou1996 commented 3 years ago

Hi @coder-people,

Thanks for asking. @yongduosui please help me with it. Thanks!

yongduosui commented 3 years ago

semisupervised_MNIST_CIFAR10 only include experiments for MNIST and CIFAR10 graph classification, not for SBMs. Please ignore main_SBMs_node_classification.py file.

Austinzhenghua commented 3 years ago

Hi @yongduosui when I run the main_superpixels_graph_classification.py --config 1 --aug 0 --load_model --head --gpu_id 0 I got this I don't know why it happed. Thanks!

yongduosui commented 3 years ago

@Austinzhenghua Please put the download MNSIT or CIFAR10 superpixel dataset on the "data/superpixels/" and try again?