diff --git a/models/architectures.py b/models/architectures.py index e84e87c..c547ff7 100644 --- a/models/architectures.py +++ b/models/architectures.py @@ -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: diff --git a/models/blocks.py b/models/blocks.py index 0f27622..f7965a7 100644 --- a/models/blocks.py +++ b/models/blocks.py @@ -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) # ---------------------------------------------------------------------------------------------------------------------- #