search

littlebeen / ASDDPM-Adaptive-Semantic-Enhanced-DDPM

The code for Adaptive Semantic-Enhanced Denoising Diffusion Probabilistic Model for Remote Sensing Image Super-Resolution
25 stars 1 forks source link

RRDB是否损坏？ #8

Open triste-creator opened 2 weeks ago

triste-creator commented 2 weeks ago

在读取权重文件rrdb.load_state_dict(torch.load('./models/LREncoder/pretrain/RRDB/model_best3.pt'), strict=False)，会RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory 权重文件是否正确？

littlebeen commented 2 weeks ago

It seems like you don't put the file into the right place or the python could not find the right path. You need to put the model_best3.pt into ./models/LREncoder/pretrain/RRDB/. It is a relative path which also related to the path you run the python file. You could change it to a absolute path such as xxxx/ASDDPM-Adaptive-Semantic-Enhanced-DDPM/models/LREncoder/pretrain/RRDB/model_best3.pt.