Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

用自制数据集进行训练,微调时RuntimeError: torch.cat(): expected a non-empty list of Tensors #19

Open
liniceyo opened this issue Oct 23, 2024 · 2 comments

Comments

@liniceyo
Copy link

微信图片_20241023174914
你好!
非常感谢你优秀的工作,有个问题想请教一下,我在使用自制数据集跑您的网络时出现了上图的问题,但是我找不到参数的设置。非常感谢您的回复!

@liniceyo
Copy link
Author

liniceyo commented Oct 24, 2024

你好!我已经找到问题的所在,修改了novel_class=(6, 7, 8),成功运行了代码。但是对您的其他参数如何设置依然感到困惑,可否指导下参数如何搭配才可以到达模型的最佳效果。如:PrototypesDistillation,PrototypesAssignment,参数应当如何设置。非常感谢您的回复!
微信图片_20241024163350

@wangchen1801
Copy link
Owner

Hello,num_queries是指每类目标分配的fine-grained prototypes数量,num_bg指背景对应数量,dim是特征维度,这三个建议用默认的就好;num_base_cls基类数量是要根据自定义数据集定的。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants