Graphormer is a general-purpose deep learning backbone for molecular modeling.
2.08k
stars
334
forks
source link
Error --AttributeError: module 'distutils' has no attribute 'version'-- when running fairseq-train and evaluate.py examples #198
Open
CESARDELATORRE opened 3 months ago
I have installed Graphormer. When I run the examples from the quickstart page, I get this error:
RUN:
fairseq-train --user-dir ../../graphormer --num-workers 16 --ddp-backend=legacy_ddp --dataset-name zinc --dataset-source pyg --task graph_prediction --criterion l1_loss --arch graphormer_slim --num-classes 1 --attention-dropout 0.1 --act-dropout 0.1 --dropout 0.0 --optimizer adam --adam-betas '(0.9, 0.999)' --adam-eps 1e-8 --clip-norm 5.0 --weight-decay 0.01 --lr-scheduler polynomial_decay --power 1 --warmup-updates 60000 --total-num-update 400000 --lr 2e-4 --end-learning-rate 1e-9 --batch-size 64 --fp16 --data-buffer-size 20 --encoder-layers 12 --encoder-embed-dim 80 --encoder-ffn-embed-dim 80 --encoder-attention-heads 8 --max-epoch 10000 --save-dir ./ckpts
ERROR:
AttributeError: module 'distutils' has no attribute 'version'
If I run the "Evaluate" Python script, I get a similar error:
RUN:
python evaluate.py --user-dir ../../graphormer --num-workers 16 --ddp-backend=legacy_ddp --dataset-name pcqm4m --dataset-source ogb --task graph_prediction --criterion l1_loss --arch graphormer_base --num-classes 1 --batch-size 64 --pretrained-model-name pcqm4mv1_graphormer_base --load-pretrained-model-output-layer --split valid --seed 1
ERROR: AttributeError: module 'distutils' has no attribute 'version'
Am I missing anything? Is module 'distutils' the right version? Thanks.