Skip to content

Commit

Permalink
problem leading to key-mismatch in pretrained model->resolved by chan…
Browse files Browse the repository at this point in the history
…ging default value of detailed_output to False
  • Loading branch information
vinits5 committed Jun 11, 2023
1 parent 2db7f8a commit 0ba6e5c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion models/pcn.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
from .pooling import Pooling

class PCN(torch.nn.Module):
def __init__(self, emb_dims=1024, input_shape="bnc", num_coarse=1024, grid_size=4, detailed_output=True):
def __init__(self, emb_dims=1024, input_shape="bnc", num_coarse=1024, grid_size=4, detailed_output=False):
# emb_dims: Embedding Dimensions for PCN.
# input_shape: Shape of Input Point Cloud (b: batch, n: no of points, c: channels)
super(PCN, self).__init__()
Expand Down

0 comments on commit 0ba6e5c

Please sign in to comment.