StyleNeRF / conf /model /stylenerf_ffhq_eg3d.yaml
Jiatao Gu
add code from the original repo
94ada0b
raw
history blame
2.41 kB
# @package _group_
name: stylenerf_ffhq
G_kwargs:
class_name: "training.networks.Generator"
z_dim: 512
w_dim: 512
mapping_kwargs:
num_layers: ${spec.map}
synthesis_kwargs:
# global settings
num_fp16_res: ${num_fp16_res}
channel_base: 1
channel_max: 512
conv_clamp: 256
kernel_size: 3
architecture: skip
z_dim: 0
resolution_vol: 128
resolution_start: 32
rgb_out_dim: 32
use_noise: False
module_name: "training.stylenerf.NeRFSynthesisNetwork"
no_bbox: True
margin: 0
magnitude_ema_beta: 0.999
camera_kwargs:
range_v: [1.4157963267948965, 1.7257963267948966]
range_u: [-0.3, 0.3]
range_radius: [1.0, 1.0]
depth_range: [0.88, 1.12]
fov: 12
gaussian_camera: True
angular_camera: True
depth_transform: ~
dists_normalized: True
ray_align_corner: False
bg_start: 0.5
renderer_kwargs:
n_ray_samples: 32
abs_sigma: False
hierarchical: True
no_background: True
foreground_kwargs:
downscale_p_by: 1
use_style: "StyleGAN2"
predict_rgb: False
use_viewdirs: False
add_rgb: True
n_blocks: 0
input_kwargs:
output_mode: 'tri_plane_reshape'
input_mode: 'random'
in_res: 4
out_res: 256
out_dim: 32
upsampler_kwargs:
no_2d_renderer: False
block_reses: ~
shared_rgb_style: False
upsample_type: "bilinear"
progressive: False
prog_nerf_only: True
# reuglarization
n_reg_samples: 0
reg_full: False
D_kwargs:
class_name: "training.stylenerf.Discriminator"
epilogue_kwargs:
mbstd_group_size: ${spec.mbstd}
num_fp16_res: ${num_fp16_res}
channel_base: ${spec.fmaps}
channel_max: 512
conv_clamp: 256
architecture: skip
progressive: False
dual_input_res: 128
upsample_type: "bilinear"
resize_real_early: True
# loss kwargs
loss_kwargs:
pl_batch_shrink: 2
pl_decay: 0.01
pl_weight: 2
style_mixing_prob: 0.9
curriculum: [0,5000]