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

when i use H_SEGFIX.json to train cityscapes datasets meet the error: #46

Closed
qingchengboy opened this issue Dec 7, 2020 · 6 comments
Closed

Comments

@qingchengboy
Copy link

In loss_heleper.py
In the calculation of loss function, the input is two tensors[1,8,128,128] /[1,2,128,128], and the corresponding label of single is three tensors.[1,512,512],[1,512,512],[1,512,512]

targets=targets_.clone().unsqueeze(1).float()
AttributeError:'list' object has no attribute 'clone'

@PkuRainBow
Copy link
Contributor

@hsfzxjy Please help to check the possible reasons.

@hsfzxjy
Copy link
Contributor

hsfzxjy commented Dec 7, 2020

Hi @qingchengboy

When training SegFix, SegFixLoss should be used. You may post the script you are using so that we can figure out the problem.

@qingchengboy
Copy link
Author

@hsfzxjy Thank you. That's the question. I chose the wrong loss type (”fs_auxce_loss“)for calculation

@hsfzxjy hsfzxjy closed this as completed Dec 8, 2020
@YAwei666
Copy link

@qingchengboy so how to change the script
of the H_SEGFIX.json?

@qingchengboy
Copy link
Author

@YAwei666
In the H_SEGFIX.json
"model_name": "segfix_hrnet"

@Jonnyboyyyy
Copy link

I am confused.
Is changing 'model_name' enough?
Do I have to change 'loss_type' too, if I want to train SegFix?

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

5 participants