Skip to content

GitLab

Unverified Commit 59e9b97f authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Merge pull request #2413 from xcyan/fix_2219 

Fix 2219 [PTN]: Increase the default batchsize from 1 to 6
parents 4347c2c5 0f83aff1
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/ptn/train_ptn.py
View file @ 59e9b97f
...@@ -44,7 +44,7 @@ flags.DEFINE_integer('image_size', 64, ...@@ -44,7 +44,7 @@ flags.DEFINE_integer('image_size', 64,
'Input images dimension (pixels) - width & height.') 'Input images dimension (pixels) - width & height.')
flags.DEFINE_integer('vox_size', 32, 'Voxel prediction dimension.') flags.DEFINE_integer('vox_size', 32, 'Voxel prediction dimension.')
flags.DEFINE_integer('step_size', 24, 'Steps to take in rotation to fetch viewpoints.') flags.DEFINE_integer('step_size', 24, 'Steps to take in rotation to fetch viewpoints.')
flags.DEFINE_integer('batch_size', 1, 'Batch size while training.') flags.DEFINE_integer('batch_size', 6, 'Batch size while training.')
flags.DEFINE_float('focal_length', 0.866, 'Focal length parameter used in perspective projection.') flags.DEFINE_float('focal_length', 0.866, 'Focal length parameter used in perspective projection.')
flags.DEFINE_float('focal_range', 1.732, 'Focal length parameter used in perspective projection.') flags.DEFINE_float('focal_range', 1.732, 'Focal length parameter used in perspective projection.')
flags.DEFINE_string('encoder_name', 'ptn_encoder', flags.DEFINE_string('encoder_name', 'ptn_encoder',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment