Wangman1

Hello, sorry to bother you. When I trained the vox model from scratch, the vgg loss was always high at the beginning, and my learning rate was 0.002. Is this...

> @Wangman1 This is my training loss. I am fixing on it... > > 请问您用您自己训练好的模型测试，能 drive 成功吗

> @Wangman1 可以的。这也是我觉得奇怪的地方，loss 看着不太对，但是可以驱动成功。 我最后也训练到和你差不多的 loss，但驱动不成功，我用了很少一部分 vox 数据来训练，不知道是不是数据的问题，你用的 htdf 数据吗 我用的这个训练命令：python train.py --dataset vox --exp_path vox_exp --exp_name 20231020 --batch_size 56 --lr 0.002，不知道有没有问题

> @Wangman1 可以的。这也是我觉得奇怪的地方，loss 看着不太对，但是可以驱动成功。 而且这个训练的很慢，大概一个礼拜训练了 10w iter，你训练的速度如何

> @Wangman1 我是先载入作者 release 的 vox 模型，然后再去 hdtf 训练的，作者应该用的是 vox1 数据集，这个数据集大概有 400 多万张图片，你用的是多少？如果太少的话确实有可能训练不成功。训练是 8 卡 A10， 10w 次迭代大概用了 1 天时间，我这里是机械硬盘，GPU 利用率还没有打满，速度上还有提升空间。 好嘞，我只用了大概 2w，这个数据集下载很慢，当时手上只有很少的数据，我是尝试从头训练的，目前还没有成功，等我多下点数据再试试

> @Wangman1 This is the loss trained on vox1 from scratch by myself , maybe it will help. 👍🏻👍🏻👍🏻 感谢您~

收到，感谢您的回复~

@yxchen1119 ，you can refer to the following modification in segmentation_distiller.py 

Thank you for your reply. Looking forward to your open source！ ------------------ 原始邮件 ------------------ 发件人: "lhwcv/mlsd_pytorch" ***@***.***>; 发送时间: 2021年7月13日(星期二) 中午11:57 ***@***.***>; ***@***.******@***.***>; 主题: Re: [lhwcv/mlsd_pytorch] How to train the models (#2) Sorry for my...

Wow, these results look great, and I can't wait to see how you train the model, so great job, thank you!!!