Open tianty opened 3 years ago
已经按照说明下载WFLW数据集,以及运行SetPreparation.py生成训练数据。但是训练在200个epoch后loss均已收敛,train_loss维持在2左右,train_loss_l2维持在0.224左右,但是test_failure_rate很高,为0.22。同时运行test_model.py发现,效果比较差,眼睛附近的landmark基本错误。 我怀疑是训练过程中出现了问题,或者是训练计划不对,请问按照train.sh中的参数learning rate等参数设置可以成功训练么,大概需要多少个epoch? 同时请问大家,如果有训练效果较好的,能否分享一下相关细节。
已经按照说明下载WFLW数据集,以及运行SetPreparation.py生成训练数据。但是训练在200个epoch后loss均已收敛,train_loss维持在2左右,train_loss_l2维持在0.224左右,但是test_failure_rate很高,为0.22。同时运行test_model.py发现,效果比较差,眼睛附近的landmark基本错误。 我怀疑是训练过程中出现了问题,或者是训练计划不对,请问按照train.sh中的参数learning rate等参数设置可以成功训练么,大概需要多少个epoch? 同时请问大家,如果有训练效果较好的,能否分享一下相关细节。