-
Notifications
You must be signed in to change notification settings - Fork 34
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
Pretrained DACLIP in options file #66
Comments
作者您好!感谢您精彩的工作,我在尝试训练代码时对options文件中opt['path']['daclip']这个路径感到有点疑惑,我如果要完全重新训练整个模型的话应该要给这个路径赋空对吧? |
为什么我将该路径置为空后还是要从huggingface上下载模型呢?请问如何从头训练模型呢? |
我好像明白了,是要先在DACLIP文件夹里训练出daclip对吧,然后再把训练得到的daclip权重与IR-SDE结合训练,是一个两阶段训练的过程 |
是的,预训练的daclip模型可以从readme里下载哈。 |
我还有一个问题,就是laion2b_s34b_b79k这个文件到底是什么呢,我看daclip以及universal IR都提到了这个文件,但是在universal IR中是在pretrained path未提供时才调用,与daclip_ViT-B-32是if else的关系,dacliP直接是pretrained path(所以我在完全重新训练时也是要置空对吧)。 |
|
@ZG-yuan Hi! The degradation loss is easier to converge, but your training is ok since the contrastive loss still decreases in the last few epochs. |
Traceback (most recent call last):
File "/opt/data/private/daclip-uir-main/universal-image-restoration/config/daclip-sde/train.py", line 356, in
main()
File "/opt/data/private/daclip-uir-main/universal-image-restoration/config/daclip-sde/train.py", line 205, in main
clip_model, preprocess = open_clip.create_model_from_pretrained('daclip_ViT-B-32', pretrained=opt['path']['daclip'])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/data/private/daclip-uir-main/universal-image-restoration/config/daclip-sde/../../open_clip/factory.py", line 379, in create_model_from_pretrained
model = create_model(
^^^^^^^^^^^^^
File "/opt/data/private/daclip-uir-main/universal-image-restoration/config/daclip-sde/../../open_clip/factory.py", line 247, in create_model
raise RuntimeError(error_str)
RuntimeError: Pretrained weights (pretrained/daclip_ViT-B-32.pt) not found for model daclip_ViT-B-32.Available pretrained tags ([].
The text was updated successfully, but these errors were encountered: