Skip to content

release lego model

Compare
Choose a tag to compare
@kwea123 kwea123 released this 20 Apr 02:47
· 195 commits to master since this release
a070d07

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