关于topk的设置问题
#1201
-
|
在dataset的配置文件中有设置evaluation的topk，用来输出和作为best_model的选择依据，然后在model的配置参数中的head字段中也有个topk的设置，但是在代码端好像没有找到用的地方，不知道模型参数中的topk的作用是什么？ |
Beta Was this translation helpful? Give feedback.
Answered by
yingfhu
Nov 15, 2022
Replies: 1 comment 1 reply
-
|
参考代码 topk in head，这里head里的topk用于在forward计算完loss之后计算当前batch的acc，需要使用 |
Beta Was this translation helpful? Give feedback.
1 reply
-
|
明白了，看漏了代码，谢谢！ |
Beta Was this translation helpful? Give feedback.
Answer selected by
ayicherry
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
参考代码 topk in head，这里head里的topk用于在forward计算完loss之后计算当前batch的acc，需要使用
cal_acc来控制。