DAMO-YOLO
DAMO-YOLO copied to clipboard
关于CWD_Loss的问题
Before Asking
-
[X] I have read the README carefully. 我已经仔细阅读了README上的操作指引。
-
[X] I want to train my custom dataset, and I have read the tutorials for finetune on your data carefully and organize my dataset correctly; 我想训练自定义数据集,我已经仔细阅读了训练自定义数据的教程,以及按照正确的目录结构存放数据集。
-
[X] I have pulled the latest code of main branch to run again and the problem still existed. 我已经拉取了主分支上最新的代码,重新运行之后,问题仍不能解决。
Search before asking
- [X] I have searched the DAMO-YOLO issues and found no similar questions.
Question
这块的代码
y_s
和y_t
的shape
都是 (N, C, H, W)
.
for idx, (s, t) in enumerate(zip(y_s, y_t)):
assert s.shape == t.shape
N, C, H, W = s.shape
通过 enumerate
拿到的 s
和 t
的 shape
应该是 (C, H, W)
. 这里的 N
是多少呢?
Additional
No response