pyg_autoscale
pyg_autoscale copied to clipboard
The code cannot exit properly
Hi, thank you very much for your extraordinary work, it helps me a lot. Now I have a problem when I run the the following command in the examples directory: python train_gcn.py --root=/my_data --device=0. There is no error raised, but the code just cannot exit properly after the training is finished, and I have to exit manually using Ctrl+C as following. I have tried to add "sys.exit(0)" in the train_gcn.py, but it doesn't work.
Traceback (most recent call last): File "/home/qchen/anaconda3/envs/py38/lib/python3.8/threading.py", line 1388, in _shutdown lock.acquire() KeyboardInterrupt:
My dependencies are python38 + pytorch 1.10.1 + pyg 2.0.3. Do you have any idea by what it is caused? Thank you again for your contribution!
Thanks for reporting. Can you do me a favor and test if the same issues persists for you on torch-geometric==1.7.2? Simply running
pip install --upgrade torch-geometric==1.7.2
and removing any pre-processed data should be sufficient to test.
I've tested following the above instructions, and the same issue persists for me.
Thanks for confirming. I will try to look into this.