Skip to content
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

How to load weights #43

Open
liboyangfaw opened this issue Nov 14, 2024 · 0 comments
Open

How to load weights #43

liboyangfaw opened this issue Nov 14, 2024 · 0 comments

Comments

@liboyangfaw
Copy link

" By default, the weights used by different methods are automatically downloaded (model.kwargs.pretrained=True, model.kwargs.checkpoint_path=''). If you prefer to specify offline weights, you can download the model weights to model/pretrain and modify the settings to model.kwargs.pretrained=False, model.kwargs.checkpoint_path='model/pretrain/xxx.pth' "

Hi, I'm trying to run MambaAD with this toolbox.

I don't konw which “model.kwargs.pretrained” and "model.kwargs.checkpoint_path" to modify, becasue there are "model_s.kwargs", "model_t.kwargs" and "model.kwargs" in configs/base /cfg_model_MambaAD, and there are "model_s.kwargs", "model_t.kwargs" and "model.kwargs" again in configs/MambaAD/MambaAD.py, which makes I feel a little confused.

I wonder 1. what do "model_s", "model_t" (pretrained encoder maybe? but I don't know the difference between t and s) and "model"(main model I guess) excatly mean? 2. which congfig do I need to change these two congfigs both? and since pretrained weights can be downloaded automatically acoording to the model name, are there any more optionsn of "model_t.name", "model_s.name" and "model.name" you may list out in the homepage or the reply.

Looking foward to your reply and I would be very grateful for that.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant