Log SummaryWritters to Tensorboard

logs/blender_paper_lego/args.txt

@@ -0,0 +1,45 @@
+N_importance = 128
+N_rand = 1024
+N_samples = 64
+basedir = ./logs
+chunk = 32768
+config = configs/lego.txt
+datadir = ./data/nerf_synthetic/lego
+dataset_type = blender
+expname = blender_paper_lego
+factor = 8
+ft_path = None
+half_res = True
+i_embed = 0
+i_img = 500
+i_print = 100
+i_testset = 50000
+i_video = 50000
+i_weights = 10000
+lindisp = False
+llffhold = 8
+lrate = 0.0005
+lrate_decay = 500
+multires = 10
+multires_views = 4
+n_gpus = 3
+netchunk_per_gpu = 262144
+netdepth = 8
+netdepth_fine = 8
+netwidth = 256
+netwidth_fine = 256
+no_batching = True
+no_ndc = False
+no_reload = False
+perturb = 1.0
+precrop_frac = 0.5
+precrop_iters = 500
+raw_noise_std = 0.0
+render_factor = 0
+render_only = False
+render_test = False
+shape = greek
+spherify = False
+testskip = 8
+use_viewdirs = True
+white_bkgd = True

logs/blender_paper_lego/config.txt

@@ -0,0 +1,19 @@
+expname = blender_paper_lego
+basedir = ./logs
+datadir = ./data/nerf_synthetic/lego
+dataset_type = blender
+
+no_batching = True
+
+use_viewdirs = True
+white_bkgd = True
+lrate_decay = 500
+
+N_samples = 64
+N_importance = 128
+N_rand = 1024
+
+precrop_iters = 500
+precrop_frac = 0.5
+
+half_res = True

run_nerf.py

@@ -691,7 +691,7 @@ def train():
     print('VAL views are', i_val)
 
     # Summary writers
-    # writer = SummaryWriter(os.path.join(basedir, 'summaries', expname))
+    writer = SummaryWriter(os.path.join(basedir, 'summaries', expname))
     
     start = start + 1
     for i in trange(start, N_iters):