search

yanghangAI / DKiS

6 stars 2 forks source link

pre-trained model #2

Open Vickie0710 opened 8 months ago

Vickie0710 commented 8 months ago

Hello, I am very interested in your work. Can you provide a pre-trained model?

yanghangAI commented 8 months ago

I have uploaded the pre-trained model.

"model_state.pt" in "final_state" folder is a pre-trained model state trained on a subset of COCO2014 which contains 6,000 images a=0.44, b=0.28 for iwt and dwt

Vickie0710 commented 8 months ago

Thank you very much for your response. I'm currently encountering an issue when trying to load the pre-trained model you provided. The error message I'm receiving is as follows: image

yanghangAI commented 8 months ago

No need to run line 25 Demo

net = DKiS(alpha_list=[0.6 ** i for i in range(16)])
net.load_state_dict(torch.load('final_state/model_state.pt'))