Method __init__

models/strand_codec.py:381–397

(self, do_vae=True, scale_init=30.0, decode_type="dir", decode_random_verts=False, nr_verts_per_strand=256, nr_values_to_decode=255, dim_per_value_decoded=3)

'''
class StrandCodec(nn.Module):
    def __init__(self, do_vae=True, scale_init=30.0, decode_type="dir", decode_random_verts=False, nr_verts_per_strand=256, nr_values_to_decode=255, dim_per_value_decoded=3):
        super(StrandCodec, self).__init__()

        self.do_vae = do_vae
        self.decode_type = decode_type
        self.decode_random_verts = decode_random_verts

        # encode
        
        self.encoder = StrandEncoder1dCNNWN(self.do_vae, out_channels=64) #Uses LinearWN
        
        # decoder
        self.decoder = StrandGeneratorSiren(in_channels=64, modulation_hidden_dim=128, siren_hidden_dim=128,
                                                     scale_init=scale_init, decode_type=decode_type, decode_random_verts=decode_random_verts,
                                                     nr_verts_per_strand=nr_verts_per_strand, nr_values_to_decode=nr_values_to_decode,
                                                     dim_per_value_decoded=dim_per_value_decoded,
                                                     )