Skip to content

Missing Encoder and Decoder Weights in Pretrained Model (params.pth) #2

Description

@kenxxxxx

Hello,

Thank you for open-sourcing the gCIS project and related resources!

While using the pretrained weights (params.pth) provided, I noticed that the weights seem to include only parameters specific to certain tasks. Based on the README, it is mentioned that "all model files share the same encoder weights," but after inspecting the contents of the params.pth file, it seems the encoder weights are not included.

My Questions:

  1. Source of Encoder Weights: Should we manually load general pretrained encoder weights? If so, could you share the exact architecture and initialization details?
  2. Complete Pretrained Model: Is there a way to obtain the full pretrained model, including encoder weights? Or is there another recommended way to get them?
  3. Loading Workflow: Could you provide a suggested workflow for loading the pretrained weights (including encoder initialization)?
    Thank you for your assistance! If there is additional documentation or guidance, please point me in the right direction.

Looking forward to your response!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions