defaulted grid_sizes in points2vols

Summary: Fix #873, that grid_sizes defaults to the wrong dtype in points2volumes code, and mask doesn't have a proper default.

Reviewed By: nikhilaravi

Differential Revision: D31503545

fbshipit-source-id: fa32a1a6074fc7ac7bdb362edfb5e5839866a472
This commit is contained in:
Jeremy Reizenstein
2021-10-16 14:40:55 -07:00
committed by Facebook GitHub Bot
parent 2f2466f472
commit 34b1b4ab8b
5 changed files with 24 additions and 4 deletions

View File

@@ -5,7 +5,7 @@
# LICENSE file in the root directory of this source tree.
import math
from typing import Tuple, Optional
from typing import Optional, Tuple
import torch
import torch.nn.functional as F