File size: 441 Bytes
8bb8404
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
model:
  base_lr: 1.0e-2
  target: renderer.renderer.RendererTrainer
  params:
    total_steps: 2000
    warm_up_steps: 100
    train_batch_num: 40960
    test_batch_num: 40960
    renderer: ngp
    cube_bound: 0.6
    use_mask: true
    lambda_rgb_loss: 0.5
    lambda_mask_loss: 10.0

data:
  target: renderer.dummy_dataset.DummyDataset
  params: {}

callbacks:
  save_interval: 5000

trainer:
  val_check_interval: 500
  max_steps: 2000