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

use pretrain model error #37

Open
myseverus opened this issue Jul 28, 2022 · 3 comments
Open

use pretrain model error #37

myseverus opened this issue Jul 28, 2022 · 3 comments

Comments

@myseverus
Copy link

Hi!I'm in trouble
I train my own datase with pre-trained models
--pretrain_model_path checkpoint.pth
the result error:
Traceback (most recent call last):
File "C:/Users/20825/Desktop/detr_code/DN-DETR-main/main.py", line 426, in
main(args)
File "C:/Users/20825/Desktop/detr_code/DN-DETR-main/main.py", line 352, in main
train_stats = train_one_epoch(
File "C:\Users\20825\Desktop\detr_code\DN-DETR-main\engine.py", line 52, in train_one_epoch
outputs = model(samples)
File "C:\anaconda\envs\Detr\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "C:\Users\20825\Desktop\detr_code\DN-DETR-main\models\DN_DAB_DETR\DABDETR.py", line 176, in forward
prepare_for_dn(dn_args, embedweight, src.size(0), self.training, self.num_queries, self.num_classes,
File "C:\Users\20825\Desktop\detr_code\DN-DETR-main\models\DN_DAB_DETR\dn_components.py", line 61, in prepare_for_dn
targets, scalar, label_noise_scale, box_noise_scale, num_patterns = dn_args
TypeError: cannot unpack non-iterable NoneType object

Thank you for your answer

@myseverus myseverus reopened this Jul 28, 2022
@myseverus
Copy link
Author

sorry,I found the reason

@jiaowanger
Copy link

What's the reason? How can I fix it

1 similar comment
@zi-boop
Copy link

zi-boop commented Apr 7, 2023

What's the reason? How can I fix it

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

3 participants