Skip to content

What is the loss after training #7

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

Open
lanyufei opened this issue Apr 30, 2020 · 5 comments
Open

What is the loss after training #7

lanyufei opened this issue Apr 30, 2020 · 5 comments

Comments

@lanyufei
Copy link

I trained with my own data set, and loss felt high and converged slowly

@Ma-Dan
Copy link
Owner

Ma-Dan commented May 1, 2020

如果是没有加载预训练权重的从头训练,请解除前面层的冻结。

@lanyufei
Copy link
Author

lanyufei commented May 1, 2020

我加载预训练权重了,最终earlystopping的loss30多,我在yolov3上训练loss训练后17多,这正常吗

@Augenstern-yzh
Copy link

我训练也遇到这种问题,loss很高,还没有我用yolov3得到的loss值低,而且loss值降低的很慢,请问这什么原因呢?

@robisen1
Copy link

I trained with my own data set, and loss felt high and converged slowly
@lanyufei did you have to use a very small batch size when training on your own dataset? I had to use 4 even though I am using a RTX 2080 TI founders edition GPU with 11g's of ram. Doing the same on qqwweee's implementation, which this borrows from, I could do 16 when training from scratch. Any idea whats using up all the memory?

@iliask97
Copy link

I trained with my own data set, and loss felt high and converged slowly
@lanyufei did you have to use a very small batch size when training on your own dataset? I had to use 4 even though I am using a RTX 2080 TI founders edition GPU with 11g's of ram. Doing the same on qqwweee's implementation, which this borrows from, I could do 16 when training from scratch. Any idea whats using up all the memory?

qqwweee's implemantation is on yolov3 not v4. yolov4 is a way bigger network and requiers a lot more vram to run. In the github repository of the original yolo their is a section witch tells you what changes you need to do according to your gpu memory to train it faster

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