We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
When setting multiple gpu training, named tuples are converted to tuples and this breaks training resulting in error
preds = self.model(batch.image) AttributeError: 'tuple' object has no attribute 'image'
Where batch is a named tuple and works fine as namedtuple when training with single gpu
The text was updated successfully, but these errors were encountered:
This may be connected to #1588
Sorry, something went wrong.
Thanks for the report! This should be resolved by PyTorch 1.8. This isn't an issue in Lightning.
A workaround in the meantime: use dictionaries instead, or extract the values from the tuple, and the first index will be the image.
No branches or pull requests
🐛 Bug
When setting multiple gpu training, named tuples are converted to tuples and this breaks training resulting in error
Where batch is a named tuple and works fine as namedtuple when training with single gpu
The text was updated successfully, but these errors were encountered: