-
Notifications
You must be signed in to change notification settings - Fork 125
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
Could you share wingloss example #31
Comments
@xialuxi @zys1994 Hi, how do you set the wing loss parameter? I have set the landmark label to [-1,1]. And I have set the wing loss parameter (w=0.3, e=2), it does not converge. As follow is the author suggestion. |
(w=10,e=2) is fine |
@xialuxi @briantse100
when i train,it makes wrong
|
你试一下使用cpu模式,看正不正常? |
cpu目前可以train,不过速度很慢 。 gpu版本是出了什么问题吗 |
等我有时间了改一下就好了。 |
@zys1994, 可以参考faster-rcnn中的smoothL1loss。 |
WingLossForward 里面这样写会存在问题,gpu多线程对同一个变量操作了。 |
将输出Dtype* out 改成, blob结构, 算完之后,再相加! |
cpu train了半天才6000个batch。期待你把gpu版本的改好来训 哈哈 |
1, 你可以使用gpu训练, 只要把loss层的gpu代码屏蔽掉就行了,训练速度影响很小。 |
make all -j 的时候报错了
这里面最后一句的one_dot_data在哪声明 |
改成one_dot.mutable_cpu_data()
…---Original---
From: "zys1994"<[email protected]>
Date: Fri, Jun 21, 2019 19:09 PM
To: "xialuxi/arcface-caffe"<[email protected]>;
Cc: "xialuxi"<[email protected]>;"Mention"<[email protected]>;
Subject: Re: [xialuxi/arcface-caffe] Could you share wingloss example (#31)
make all -j 的时候报错了
src/caffe/layers/wing_loss_layer.cpp:28:47: error: ‘one_dot_data’ was not declared in this scope void WingLossLayer<Dtype>::LayerSetUp(const vector<Blob<Dtype>*>& bottom, const vector<Blob<Dtype>*>& top) { w = this->layer_param_.wing_loss_param().w(); DCHECK(w > 0); epsilon = this->layer_param_.wing_loss_param().epsilon(); DCHECK(epsilon >= 0); //_c = w * (1.0 - log(1.0 + w/epsilon) _c = w * (1.0 - log(1.0 + w/epsilon)); // has_ignore_label_ = this->layer_param_.loss_param().has_ignore_label(); // if (has_ignore_label_) // { // ignore_label_ = this->layer_param_.loss_param().ignore_label(); // } log_abs.ReshapeLike(*bottom[0]); caffe_set(bottom[0]->count(), Dtype(1.0), one_dot_data); }
这里面最后一句的one_dot_data在哪声明
—
You are receiving this because you were mentioned.
Reply to this email directly, view it on GitHub, or mute the thread.
|
改了之后make可以通过, 但用cpu,gpu 在train都会报错
|
@xialuxi 目前在用wingloss train的时候报错可以帮忙看看吗? |
Could you share wingloss in train prototxt example like EuclideanLoss
The text was updated successfully, but these errors were encountered: