-
Notifications
You must be signed in to change notification settings - Fork 673
New issue
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
There are questions about the parameters needed for forward propagation in model transformation #926
Comments
Here is my conversion script, based on the previous question, I tried to enter the parameters needed for the forward channel during the conversion:
But the following error was reported:
|
Hi @huangshilong911 , Thanks for reaching out again. It looks like the error occurs during the execution of the PyTorch model. Do you experience this error simply by calling the following? output = SlimSAM_model(*[batched_input, multimask_output]) Best, |
# for free
to join this conversation on GitHub.
Already have an account?
# to comment
Hello john, once again I have to ask for your advice sincerely.
In a previous demo, I was able to successfully convert a .pth to a .engine model with the following code:
However, while performing the conversion against the slim_sam model, the need for a forward channel input parameter arose and I had to modify the code to the following form:
But even so, the script still doesn't run well, it still prompts me that I did parameter or entered the wrong parameter, is there any way to solve this situation?
In addition, I referred to the work you recommended to me on nanosam, which also seems to add parameters to the conversion process, which seems to be informative, and I would like to ask for advice on whether I should operate the same way as this work. If so, what should I do?
The relevant operations for the nanosam work are as follows:
The text was updated successfully, but these errors were encountered: