Spaces:
Sleeping
Sleeping
File size: 478 Bytes
fe3e74d |
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 |
model:
base_lr: 5.0e-4
target: renderer.renderer.RendererTrainer
params:
total_steps: 2000
warm_up_steps: 100
train_batch_num: 3584
train_batch_fg_num: 512
test_batch_num: 4096
use_mask: true
lambda_rgb_loss: 0.5
lambda_mask_loss: 1.0
lambda_eikonal_loss: 0.1
use_warm_up: true
data:
target: renderer.dummy_dataset.DummyDataset
params: {}
callbacks:
save_interval: 500
trainer:
val_check_interval: 500
max_steps: 2000
|