4
0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 18:44:21 +08:00
nerfstudio/tests/nerf/graph/configs/test_basic_graph.yaml

27 lines
472 B
YAML

machine_config:
num_gpus: 8
num_machines: 1
machine_rank: 0
dist_url: auto
network:
mlp_0:
class_name: MLP
inputs: ['encoder_0']
meta_data:
in_dim: 0
out_dim: 1
num_layers: 2
layer_width: 3
encoder_0:
class_name: ScalingAndOffset
inputs: ['x']
meta_data:
in_dim: 16
mlp_1:
class_name: MLP
inputs: ['mlp_0']
meta_data:
in_dim: 4
out_dim: 5
num_layers: 6
layer_width: 7