Open xin1106 opened 3 months ago
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]
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]