Open edwardpwtsoi opened 5 years ago
请问能描述一下怎么样从FeatherNet的输出向量(长度1024的)计算出score吗?
参考测试的代码
_, pred = output.topk(maxk, 1, True, True) line 393 main.py
这一部分是不是出了错?topk的第二个return是indices
假如我继续运行
python main.py --config="cfgs/FeatherNetB-32.yaml" --resume ./our_pretrained_models/FeatherNetB_bs32/_47_best.pth.tar --val True --val-save True
我会得到这个报错
Traceback (most recent call last):
File "main.py", line 405, in
用breakpoint()调查了一下发现 correct = pred.eq(target.view(1, -1).expand_as(pred)) line394 main.py 这部分在比较label跟indices
思路上没有错。indices也是从0开始,对应了标签的0,1
请问能描述一下怎么样从FeatherNet的输出向量(长度1024的)计算出score吗?