Skip to content

GitLab

Commit 0f83aff1 authored by Xinchen Yan's avatar Xinchen Yan
Browse files

fix #2219: increase batchsize from 1 to 6

parent 6e4bbb74
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/ptn/train_ptn.py
View file @ 0f83aff1
...@@ -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