Commit 9429fe83 authored by yan.yan's avatar yan.yan
Browse files

add a assert for inverse conv

parent 322353c5
...@@ -346,6 +346,9 @@ class SparseConvolution(SparseModule): ...@@ -346,6 +346,9 @@ class SparseConvolution(SparseModule):
mask_argsort_bwd_splits = datas.mask_argsort_fwd_splits mask_argsort_bwd_splits = datas.mask_argsort_fwd_splits
masks = datas.masks masks = datas.masks
out_spatial_shape = datas.spatial_shape out_spatial_shape = datas.spatial_shape
assert pair_fwd.shape[0] == np.prod(
self.kernel_size
), "inverse conv must have same kernel size as its couple conv"
else: else:
if self.indice_key is not None and datas is not None: if self.indice_key is not None and datas is not None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment