DAMO-YOLO icon indicating copy to clipboard operation
DAMO-YOLO copied to clipboard

关于CWD_Loss的问题

Open JinYu1998 opened this issue 1 year ago • 0 comments

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_sy_tshape都是 (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 拿到的 stshape 应该是 (C, H, W). 这里的 N是多少呢?

Additional

No response

JinYu1998 avatar Aug 16 '23 08:08 JinYu1998