release lego model
lego model and reconstructed mesh.
Command:
python train.py \
--dataset_name blender \
--root_dir /home/ubuntu/data/nerf_example_data/nerf_synthetic/lego/ \
--N_importance 64 --img_wh 400 400 --noise_std 0 \
--batch_size 1024 --num_epochs 16 \
--optimizer adam --lr 5e-4 \
--lr_scheduler steplr --decay_step 2 4 8 --decay_gamma 0.5 \
--exp_name exp3
Detailed profile:
Action | Mean duration (s) | Total time (s)
-----------------------------------------------------------------
on_train_start | 1.2281e-05 | 1.2281e-05
on_epoch_start | 6.1691e-06 | 9.8706e-05
get_train_batch | 0.00023678 | 59.198
on_batch_start | 4.4245e-06 | 1.1061
model_forward | 0.041729 | 1.0432e+04
model_backward | 0.046964 | 1.1741e+04
on_after_backward | 1.5339e-06 | 0.38347
optimizer_step | 0.0035952 | 898.81
on_batch_end | 4.1799e-06 | 1.045
on_epoch_end | 5.4906e-06 | 8.785e-05
on_train_end | 9.583e-06 | 9.583e-06