nerf_simple.json 774 Bytes
Newer Older
Nianchen Deng's avatar
sync    
Nianchen Deng committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
{
    "model": ["NeRF", {
        "color": "rgb",
        "sampler": {
            "range": [0, 10],
            "n_samples": 256
        },
        "encoder_x": ["Freq", {
            "freqs": 10,
            "cat_input": true
        }],
        "encoder_d": ["Freq", {
            "freqs": 4,
            "angular": true
        }],
        "field": {
            "shape": [8, 256],
            "skips": [4]
        },
        "color_decoder": "NeRF",
        "cascade": false
    }],
    "train": ["Trainer", {
        "max_epochs": 20,
        "opti": ["Adam", {
            "lr": 5e-4
        }],
        "loss": {
            "Color_L2": "mse_loss(color, gt_color)"
            //"Density_Reg": ["cauchy_loss(energies)", {"weight": 1e-4, "s": 1e4}]
        }
    }]
}