Spaces:
Paused
Paused
File size: 441 Bytes
02afb14 |
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
|