mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 11:52:50 +08:00
47d06c8924
Summary: Implements a ViewPooler that groups ViewSampler and FeatureAggregator. Reviewed By: shapovalov Differential Revision: D35852367 fbshipit-source-id: c1bcaf5a1f826ff94efce53aa5836121ad9c50ec
31 lines
800 B
YAML
31 lines
800 B
YAML
defaults:
|
|
- repro_multiseq_base.yaml
|
|
- repro_feat_extractor_normed.yaml
|
|
- _self_
|
|
generic_model_args:
|
|
chunk_size_grid: 32000
|
|
num_passes: 1
|
|
n_train_target_views: -1
|
|
loss_weights:
|
|
loss_rgb_mse: 200.0
|
|
loss_prev_stage_rgb_mse: 0.0
|
|
loss_mask_bce: 1.0
|
|
loss_prev_stage_mask_bce: 0.0
|
|
loss_autodecoder_norm: 0.0
|
|
depth_neg_penalty: 10000.0
|
|
raysampler_args:
|
|
n_rays_per_image_sampled_from_mask: 2048
|
|
min_depth: 0.05
|
|
max_depth: 0.05
|
|
scene_extent: 0.0
|
|
n_pts_per_ray_training: 1
|
|
n_pts_per_ray_evaluation: 1
|
|
stratified_point_sampling_training: false
|
|
stratified_point_sampling_evaluation: false
|
|
renderer_class_type: LSTMRenderer
|
|
implicit_function_class_type: SRNImplicitFunction
|
|
view_pooler_enabled: true
|
|
solver_args:
|
|
breed: adam
|
|
lr: 5.0e-05
|