LeiPK

python train_demo.py \ --val val_wiki \ --model snail \ --encoder cnn \ --trainN 10 \ --N 5 \ --K 1 \ --val_step 1000 模型报错如下： ../aten/src/ATen/native/cuda/Indexing.cu:1141: indexSelectLargeIndex: block: [18,0,0], thread: [123,0,0]...