Skip to content

Commit 0f83aff

Browse files
committed
fix tensorflow#2219: increase batchsize from 1 to 6
1 parent 6e4bbb7 commit 0f83aff

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

research/ptn/train_ptn.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@
4444
'Input images dimension (pixels) - width & height.')
4545
flags.DEFINE_integer('vox_size', 32, 'Voxel prediction dimension.')
4646
flags.DEFINE_integer('step_size', 24, 'Steps to take in rotation to fetch viewpoints.')
47-
flags.DEFINE_integer('batch_size', 1, 'Batch size while training.')
47+
flags.DEFINE_integer('batch_size', 6, 'Batch size while training.')
4848
flags.DEFINE_float('focal_length', 0.866, 'Focal length parameter used in perspective projection.')
4949
flags.DEFINE_float('focal_range', 1.732, 'Focal length parameter used in perspective projection.')
5050
flags.DEFINE_string('encoder_name', 'ptn_encoder',

0 commit comments

Comments
 (0)