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

Orca: what if our model takes a list of tensors as one of the args as input in TorchRunner #5762

Closed
leonardozcm opened this issue Sep 14, 2022 · 0 comments
Assignees
Labels
customized training orca customized training loop support orca

Comments

@leonardozcm
Copy link
Contributor

In my case, the model does take x as a list of two tensors as input:

    def forward(self, x, bboxes=None):
        x = x[:]  # avoid pass by reference
        x = self.s1(x)

code
but our torchrunner will extract this as two separated ones:
https://github.com/intel-analytics/BigDL/blob/affe54803c320afd4fc0631dc3fa02f8be1cfcdc/python/orca/src/bigdl/orca/learn/pytorch/training_operator.py#L279

@leonardozcm leonardozcm changed the title Orca: what if our model takes a list of tensors as one of the args as input in TorchRunner Orca: what if our model takes ***a list of tensors*** as one of the args as input in TorchRunner Sep 14, 2022
@leonardozcm leonardozcm changed the title Orca: what if our model takes ***a list of tensors*** as one of the args as input in TorchRunner Orca: what if our model takes **a list of tensors** as one of the args as input in TorchRunner Sep 14, 2022
@leonardozcm leonardozcm changed the title Orca: what if our model takes **a list of tensors** as one of the args as input in TorchRunner Orca: what if our model takes a list of tensors as one of the args as input in TorchRunner Sep 14, 2022
@leonardozcm leonardozcm self-assigned this Sep 14, 2022
@leonardozcm leonardozcm added the customized training orca customized training loop support label Sep 15, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
customized training orca customized training loop support orca
Projects
None yet
Development

No branches or pull requests

1 participant