Errors occur when running protonet.py

[xin1106]

args=(--phase 'prototrain' --dataset "${DATASET}" --cvfold $SPLIT
--data_path "$DATA_PATH" --save_path "$SAVE_PATH"
--pretrain_checkpoint_path "$PRETRAIN_CHECKPOINT" --use_attention
--n_subprototypes $N_SUBPROTOTYPES --k_connect $K_CONNECT
--dist_method "$SIM_FUNCTION" --sigma $SIGMA
--pc_npts $NUM_POINTS --pc_attribs "$PC_ATTRIBS" --pc_augm
--edgeconv_widths "$EDGECONV_WIDTHS" --dgcnn_k $K
--dgcnn_mlp_widths "$MLP_WIDTHS" --base_widths "$BASE_WIDTHS"
--n_iters $NUM_ITERS --eval_interval $EVAL_INTERVAL --batch_size 1
--lr $LR --step_size $DECAY_STEP --gamma $DECAY_RATIO
--n_way $N_WAY --k_shot $K_SHOT --n_queries $N_QUESIES --n_episode_test $N_TEST_EPISODES)
When running the above train_attMPTI. sh
RuntimeError: Expected target size [2, 192], got [2, 2048]