TypeError: scatter_mean() takes from 2 to 5 positional arguments but 6 were given

[foxhxer]

Hi @Richarizardd I was trying to train GCN model but facing TypeError: scatter_mean() takes from 2 to 5 positional arguments but 6 were given

while trying to run the following command. python train_cv.py --exp_name surv_15_rnaseq --task surv --mode graph --model_name graph --niter 0 --niter_decay 50 --lr 0.002 --init_type max --reg_type none --lambda_reg 0 -use_vgg_features 1 --gpu_ids 0

Installation : torch-cluster-1.6.1 torch-geometric-1.3.0 torch-scatter-2.1.1 torch-sparse-0.6.17

Can you please suggest.

[chenxinli001]

meet same problem