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.

[XGGNet]

meet same problem