Initial commit
This commit is contained in:
parent
755317d83a
commit
c131638fa7
|
@ -53,7 +53,6 @@ class KPCNN(nn.Module):
|
|||
break
|
||||
|
||||
# Apply the good block function defining tf ops
|
||||
print(block, r)
|
||||
self.block_ops.append(block_decider(block,
|
||||
r,
|
||||
in_dim,
|
||||
|
@ -331,7 +330,6 @@ class KPFCNN(nn.Module):
|
|||
if block_i in self.encoder_skips:
|
||||
skip_x.append(x)
|
||||
x = block_op(x, batch)
|
||||
print(block_op)
|
||||
|
||||
for block_i, block_op in enumerate(self.decoder_blocs):
|
||||
if block_i in self.decoder_concats:
|
||||
|
|
|
@ -356,7 +356,9 @@ class KPConv(nn.Module):
|
|||
return torch.sum(kernel_outputs, dim=0)
|
||||
|
||||
def __repr__(self):
|
||||
return "KPConv(radius: %.2f, in_feat: %i, out_feat: %i)"% (self.radius, self.in_channels, self.out_channels)
|
||||
return 'KPConv(radius: {:.2f}, in_feat: {:d}, out_feat: {:d})'.format(self.radius,
|
||||
self.in_channels,
|
||||
self.out_channels)
|
||||
|
||||
# ----------------------------------------------------------------------------------------------------------------------
|
||||
#
|
||||
|
|
Loading…
Reference in a new issue