-
Notifications
You must be signed in to change notification settings - Fork 227
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
训练时间 #118
Comments
从提供的pretrain训练半天到一天时间能训到95%以上的性能,训得越久效果越好。从头训可能要一个月时间。
Get Outlook for Android
|
期待Training guidance~~ |
作者您好,我在您回答的#115问题中看到您说根据您提供的pre-trained backbone来训练大概需要半天,这里我不是很懂,我想问pre-trained backbone与pre-trained model具体有什么区别?相对于pre-trained model,pre-trained backbone中哪部分是没经过训练的? |
pre-trained backbone 采用的不一样的head 来学习2D pose estimation之类的其他任务,预训练出来的模型,因为在相似的任务上pretrain过,所以会收敛的比较快。 pre-trained model是指的是我训练好的模型。从0开始训一般采用例如ImageNet上训练的参数,需要比较长的时间构建基础特征。我这里提供的pre-trained backbone就是帮大家节省时间,跳过基础特征构建部分。 |
作者您好,关于训练我有个问题想问您,下面是我的训练log。 master_batch_size: -1 fine_tune: False model_return_loss: False merge_smpl_camera_head: False backbone: 'hrnet' model_path: /data01/wyjh/ROMP/trained_models(pretrained backboon)/pretrain_hrnet.pkl loss_weight: sample_prob: |
还是建议您使用pretraining模型来减少训练时间,而且大batch size对于pretraining还是很重要的,不然真的要训很久很久的。您可以等一两个周再来看结果,估计就能下降到和使用pretraining模型训练差不多的loss了。 |
那您看我这样写是否可以调用起pretrain backbone: |
您把hrnet_pretrain的路径设置好了,然后自动调用backbone就会在这里加载预训练参数用于训练。 |
作者您好,我想就训练时间问题再问您一点问题: |
1.目前训练速度确实比较慢,我在4块P40上一个batch(64)大概1.72,不太清楚在1080Ti上是什么速度。 |
作者您好,关于继续训练我有一点疑惑,在调用之前的pkl文件时,我出现了如下提醒,我觉得是有很多模块的数据找不到,是因为我没有正确调用文件,还是因为每当有效果更好的模型产生时我手动删除效果较差的模型而导致读取之前的模型数据失败呢?或是说本来就是如此? |
请问作者使用4张p40gpu训练完整的数据集需要多久?
The text was updated successfully, but these errors were encountered: