-
Notifications
You must be signed in to change notification settings - Fork 1.2k
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
Monodepth Regularization #2284
Monodepth Regularization #2284
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for finding that bug Ethan! I looked over the other changes you made and they all look good. I think it's finally ready to be merged into main.
LGTM! |
mask: Optional[Tensor] = None, | ||
device: Union[torch.device, str] = "cpu", | ||
) -> Int[Tensor, "batch_size 3"]: | ||
if mask: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think this should be: if mask is None
. My transforms.json
contains masks and as a result the code fails when maks
tensor does not evaluate as a boolean expression.
Maybe it would be possible to add instructions to the NotImplementedError
message on what needs to be adjusted in the configuration for the code to work with mask images (my case)? Naybe automatically fall back to the PixelSampler, if that's the solution?
I created this PR to resolve the GitHub actions issues from PRs #2252 and #2066, which were duplicates. See PR #2066 for a detailed discussion and the contributions from @AdamRashid96.
This line change is particularly important because we call setup() with kwargs elsewhere. We rely on being able to change the default
num_rays_per_batch
in thetest_train.py
script, and the small value of 4 for testing was never actually passed through. I believe this lead to an out of memory issue and therefore GitHub actions failing with the non-descriptive error ofError: The operation was canceled
.