预训练模型的问题

huang229 / auto_tooth_arrangement

Tooth arrangement, Medical orthodontics，Neural networks, Deep learning, Transformer, Pytorch, Python

32 stars 8 forks source link

Open HwwwwwwwH opened 11 months ago

HwwwwwwwH commented 11 months ago

请问预训练是网络的哪一部分？另外model.pth模型的链接好像没了...

huang229 commented 11 months ago

可以不用预训练模型，直接跑训练，把优化器改为Adam

huang229 commented 11 months ago

因为一些原因，模型文件被误删了，我也没做这个项目几个月了。直接训练，用训练数据做测试，就可以知道模型到底有没有排牙能力。