mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 11:52:50 +08:00
6275283202
Summary: Make dataset type and args configurable on JsonIndexDatasetMapProvider. Reviewed By: davnov134 Differential Revision: D36666705 fbshipit-source-id: 4d0a3781d9a956504f51f1c7134c04edf1eb2846
86 lines
2.1 KiB
YAML
86 lines
2.1 KiB
YAML
defaults:
|
|
- default_config
|
|
- _self_
|
|
exp_dir: ./data/exps/base/
|
|
architecture: generic
|
|
visualize_interval: 0
|
|
visdom_port: 8097
|
|
data_source_args:
|
|
dataset_map_provider_class_type: JsonIndexDatasetMapProvider
|
|
data_loader_map_provider_SequenceDataLoaderMapProvider_args:
|
|
batch_size: 10
|
|
dataset_len: 1000
|
|
dataset_len_val: 1
|
|
num_workers: 8
|
|
images_per_seq_options:
|
|
- 2
|
|
- 3
|
|
- 4
|
|
- 5
|
|
- 6
|
|
- 7
|
|
- 8
|
|
- 9
|
|
- 10
|
|
dataset_map_provider_JsonIndexDatasetMapProvider_args:
|
|
dataset_root: ${oc.env:CO3D_DATASET_ROOT}
|
|
n_frames_per_sequence: -1
|
|
test_on_train: true
|
|
test_restrict_sequence_id: 0
|
|
dataset_JsonIndexDataset_args:
|
|
load_point_clouds: false
|
|
mask_depths: false
|
|
mask_images: false
|
|
generic_model_args:
|
|
loss_weights:
|
|
loss_mask_bce: 1.0
|
|
loss_prev_stage_mask_bce: 1.0
|
|
loss_autodecoder_norm: 0.01
|
|
loss_rgb_mse: 1.0
|
|
loss_prev_stage_rgb_mse: 1.0
|
|
output_rasterized_mc: false
|
|
chunk_size_grid: 102400
|
|
render_image_height: 400
|
|
render_image_width: 400
|
|
num_passes: 2
|
|
implicit_function_NeuralRadianceFieldImplicitFunction_args:
|
|
n_harmonic_functions_xyz: 10
|
|
n_harmonic_functions_dir: 4
|
|
n_hidden_neurons_xyz: 256
|
|
n_hidden_neurons_dir: 128
|
|
n_layers_xyz: 8
|
|
append_xyz:
|
|
- 5
|
|
latent_dim: 0
|
|
raysampler_AdaptiveRaySampler_args:
|
|
n_rays_per_image_sampled_from_mask: 1024
|
|
scene_extent: 8.0
|
|
n_pts_per_ray_training: 64
|
|
n_pts_per_ray_evaluation: 64
|
|
stratified_point_sampling_training: true
|
|
stratified_point_sampling_evaluation: false
|
|
renderer_MultiPassEmissionAbsorptionRenderer_args:
|
|
n_pts_per_ray_fine_training: 64
|
|
n_pts_per_ray_fine_evaluation: 64
|
|
append_coarse_samples_to_fine: true
|
|
density_noise_std_train: 1.0
|
|
view_pooler_args:
|
|
view_sampler_args:
|
|
masked_sampling: false
|
|
image_feature_extractor_ResNetFeatureExtractor_args:
|
|
stages:
|
|
- 1
|
|
- 2
|
|
- 3
|
|
- 4
|
|
proj_dim: 16
|
|
image_rescale: 0.32
|
|
first_max_pool: false
|
|
solver_args:
|
|
breed: adam
|
|
lr: 0.0005
|
|
lr_policy: multistep
|
|
max_epochs: 2000
|
|
momentum: 0.9
|
|
weight_decay: 0.0
|