一个小小的错误

[sorrowyn]

老师模型

python3 train_B_part.py --data /media/mlldiskSSD/MLICdataset/COCO2014/ --dataset coco --gpu 2 --part 0 --num-classes 80 --subnum 2 --typ cluster --model 101 --metric mse -b 32 --lr 1e-4
python3 train_B_part.py --data /media/mlldiskSSD/MLICdataset/COCO2014/ --dataset coco --gpu 3 --part 0 --num-classes 80 --subnum 2 --typ coocurrence --model 101 --metric mse -b 32 --lr 1e-4

学生模型

python3 dis_B_cpsd.py --data /media/mlldiskSSD/MLICdataset/COCO2014/ --dataset coco -t 101 -s 101 --num-classes 80 --subnum 2 --model-root ./checkpoints/coco/101/subnum2 --metric mse
下面这部分代码出错

for eachidxdic in idx_cl:
    sort_idx_cl += list(eachidxdic.keys())            
for j, each in enumerate(sort_idx_cl):
    dis_target_cl[:, int(each)] = output_t_cl[:, j]

[Robbie-Xu]

老师模型

python3 train_B_part.py --data /media/mlldiskSSD/MLICdataset/COCO2014/ --dataset coco --gpu 2 --part 0 --num-classes 80 --subnum 2 --typ cluster --model 101 --metric mse -b 32 --lr 1e-4
python3 train_B_part.py --data /media/mlldiskSSD/MLICdataset/COCO2014/ --dataset coco --gpu 3 --part 0 --num-classes 80 --subnum 2 --typ coocurrence --model 101 --metric mse -b 32 --lr 1e-4

学生模型

python3 dis_B_cpsd.py --data /media/mlldiskSSD/MLICdataset/COCO2014/ --dataset coco -t 101 -s 101 --num-classes 80 --subnum 2 --model-root ./checkpoints/coco/101/subnum2 --metric mse
下面这部分代码出错

for eachidxdic in idx_cl:
    sort_idx_cl += list(eachidxdic.keys())            
for j, each in enumerate(sort_idx_cl):
    dis_target_cl[:, int(each)] = output_t_cl[:, j]

谢谢提供的bug信息，目前没有进一步维护该代码，之后有空的时候会复现该错误并更正。