(self, in_features, out_features, bias=True)

( self, in_channels, out_channels, kernel_size, stride=1, padd

(self, num_channels, bandwidth=1)

(self)

(self, crop_size,top_ratio, left_ratio)

(self, inner_model, sigma_data=1., weighting='karras', scales=1, parametrization="v", loss_weight_per_channel=

(self, *main, skip=None)

(self, module)

(self, *main, skip=None)

(self, feats_in, c_out, num_groups, eps=1e-5, cond_key='cond')

(self, c_in, n_head, norm, dropout_rate=0.)

(self, c_dec, c_enc, n_head, norm_dec, dropout_rate=0., cond_key='cross', cond_key_padding='c

(self, kernel='linear', pad_mode='reflect')

(self, kernel='linear', pad_mode='reflect')

(self, in_features, out_features, std=1.)

(self, optimizer, inv_gamma=1., power=1., warmup=0., min_lr=0., last_epoch=-1, verbose=False)

(self, optimizer, num_steps, decay=0.5, warmup=0., min_lr=0., last_epoch=-1, verbose=False)

(self, optimizer, warmup=0., last_epoch=-1, verbose=False)

(self, root, transform=None)

(self, filename, columns)

(self, x, t0, t1, seed=None, **kwargs)

(self, x, sigma_min, sigma_max, seed=None, transform=lambda x: x)

(self, h, pcoeff, icoeff, dcoeff, order=1, accept_safety=0.81, eps=1e-8)

(self, model, extra_args=None, eps_callback=None, info_callback=None)

(self)

(self, function, *args, **kwargs)

(self, levels, mapping, in_channels, out_channels, patch_size, input_size, condition_dropout_rate, rgb_conditi

(self, dim, n_heads, start_index=0, freqs_init=freqs_pixel_log(max_freq=10.0))

(self, in_features, out_features, kernel_size, bias=True)

(self, in_features, out_features, bias=True)

(self, features, cond_features, eps=1e-6)

(self, dim, n_heads)

(self, d_model, d_head, cond_features, dropout=0.0)

(self, d_model, d_head, cond_features, kernel_size, dropout=0.0)

(self, d_model, d_head, cond_features, window_size, window_shift, dropout=0.0)

(self, d_model, d_ff, cond_features, dropout=0.0)

(self, d_model, d_ff, d_head, cond_features, dropout=0.0)

(self, d_model, d_ff, d_head, cond_features, kernel_size, dropout=0.0)

(self, d_model, d_ff, d_head, cond_features, window_size, index, dropout=0.0)

(self, d_model, d_ff, cond_features, dropout=0.0)

(self, d_model, d_ff, dropout=0.0)

(self, n_layers, d_model, d_ff, dropout=0.0)

(self, in_features, out_features, patch_size=(2, 2))

(self, in_features, out_features, patch_size=(2, 2))

(self, in_features, out_features, patch_size=(2, 2))

(self, input_dim, out_dim, cond_dims)

(self, in_channels, n_heads, d_head, global_cond_dim, do_self_attention=True

(self, npz_data, meta_data, scalp_mesh_data, nr_full_strands_per_hairstyle, compute_tbn=True, flip=False)

(self, hairstyle_path)

(self, difflocks_path, processed_difflocks_path=None, #contains path to scal

(self)

(self, viewer_config_path, experiment_name)

(self, experiment_name, config_path, entity)

(self, callbacks)

(self, experiment_name)

(self, name, loader, grad)

(self)

(self, path_ckpt_strandcodec, path_config_difflocks, path_ckpt_difflocks, path_ckpt_rgb2material=None, cfg_val

(self, optimizer, multiplier, total_epoch, after_scheduler=None)

(self, optimizer, start_factor=1.0 / 3, end_factor=1.0, total_iters=5, last_epoch=-1, verbose

(self, optimizer, milestones, gamma=0.1, last_epoch=-1, verbose=False)

(self, optimizer, last_step=-1)

(self, optimizer, last_step=-1)

(self, optimizer, warmup_period, last_step=-1)

(self, optimizer, warmup_period, last_step=-1)

(self, do_vae, out_channels=128, num_pts=256)

(self, in_channels, modulation_hidden_dim, siren_hidden_dim, scale_init, decode_type, decode_random_verts, nr_

(self, input_dim, out_dim, hidden_dim)

( self, eps: float = 1e-8, )

(self)

(self)

(self)

(self)

(self)

(self, sigma)

(self, sigma)

(self, sigma)

(self, **kwargs)

(self, phase, gt_cloud=None, pred_cloud=None, **kwargs)

(self, phase, loss, loss_kl, **kwargs)

(self, **kwargs)

(self, phase, loss=0, loss_pos=0, loss_dir=0, loss_curv=0, loss_kl=0, lr=0, z_deviation=None, z=None, z_no_eps

(self, phase, loss, loss_pos, loss_dir, loss_curv, **kwargs)

(pred_dict, normalization_dict)

(x, theta, conj=False)