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

Multi-parameter input problem #396

Open
IAmAlwaysOnline opened this issue Jul 1, 2024 · 1 comment
Open

Multi-parameter input problem #396

IAmAlwaysOnline opened this issue Jul 1, 2024 · 1 comment

Comments

@IAmAlwaysOnline
Copy link

IAmAlwaysOnline commented Jul 1, 2024

So my example_input is two tensors how should I pass in, so I have one input and one auxiliary input
dummy_input_x = torch.randn(1, IN_CHANNEL, 224, 224).to(device)
dummy_input_x_ex = torch.randn(1, EX_CHANNEL, 224, 224).to(device)
Can I spell tuple or list, but the assertion error is triggered when run: RuntimeError: CUDA error: device-side assert triggered

@Edgar454
Copy link

Edgar454 commented Jul 1, 2024

Use a dictionnary to specify the the field and inputs associated
for example : example_inputs = {"field1": dummy_input_x , "field2":dummy_input_x_ex }

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

2 participants