Does the code provide a pretrained model for Encoder？

zhuqinfeng1999 / Seg-LSTM

Apache License 2.0

12 stars 1 forks source link

Does the code provide a pretrained model for Encoder？ #2

Open yangchengbang opened 1 month ago

yangchengbang commented 1 month ago

I want to use Encoder as the backbone network to extract features. Does the code have a pretrained model for Encoder？

CODERED555 commented 1 month ago

Hi,may I ask you aquestion about log_fg_matrix = torch.where(ltr, _log_fg_matrix[:, :, 1:, 1:], -float("inf")) # (B, NH, S, S) RuntimeError: expected scalar type float but found double have you met this problem